Fix bugs

2021-03-28 22:26:59 -07:00
parent 62bedaa094
commit c12601bc0b
1 changed files with 1 additions and 1 deletions
--- a/lib/trade_models/quant_transformer.py
+++ b/lib/trade_models/quant_transformer.py
@@ -298,7 +298,7 @@ class QuantTransformer(Model):
                results_dict=results_dict,
                start_epoch=iepoch + 1,
            )
-            torch.save(save_info, ckp_path, map_location="cpu")
+            torch.save(save_info, ckp_path)
        self.logger.info(
            "The best score: {:.6f} @ {:02d}-th epoch".format(best_score, best_epoch)
        )