iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bugs

Browse Source
This commit is contained in:
D-X-Y
2021-03-28 22:26:59 -07:00
parent 62bedaa094
commit c12601bc0b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/trade_models/quant_transformer.py
View File

@@ -298,7 +298,7 @@ class QuantTransformer(Model):
results_dict=results_dict, results_dict=results_dict,
start_epoch=iepoch + 1, start_epoch=iepoch + 1,
) )
torch.save(save_info, ckp_path, map_location="cpu") torch.save(save_info, ckp_path)
self.logger.info( self.logger.info(
"The best score: {:.6f} @ {:02d}-th epoch".format(best_score, best_epoch) "The best score: {:.6f} @ {:02d}-th epoch".format(best_score, best_epoch)
) )