From 6b90c6d066d313f0380c3a537769e977d7d0c4e6 Mon Sep 17 00:00:00 2001 From: Alex Wang Date: Wed, 25 Nov 2020 14:23:12 +0800 Subject: [PATCH] update --- qlib/contrib/model/pytorch_sfm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_sfm.py b/qlib/contrib/model/pytorch_sfm.py index cf33732b9..1d3012331 100644 --- a/qlib/contrib/model/pytorch_sfm.py +++ b/qlib/contrib/model/pytorch_sfm.py @@ -398,7 +398,7 @@ class SFM(Model): # update learning rate self.scheduler.step(cur_loss_val) - if device != 'cpu': + if self.device != 'cpu': torch.cuda.empty_cache() def get_loss(self, pred, target, loss_type):