Fix bugs

2021-03-28 22:26:59 -07:00 · 2021-03-28 22:26:59 -07:00 · c12601bc0b
commit c12601bc0b
parent 62bedaa094
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)
        )