liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 16:26:55 +08:00
Code Activity

update

Browse Source
This commit is contained in:
Alex Wang
2020-11-25 14:23:12 +08:00
parent dad09e91cf
commit 6b90c6d066
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_sfm.py
View File

@@ -398,7 +398,7 @@ class SFM(Model):
# update learning rate # update learning rate
self.scheduler.step(cur_loss_val) self.scheduler.step(cur_loss_val)
if device != 'cpu': if self.device != 'cpu':
torch.cuda.empty_cache() torch.cuda.empty_cache()
def get_loss(self, pred, target, loss_type): def get_loss(self, pred, target, loss_type):