Open caolongjie1101 opened 9 months ago
@you-n-g can I work on this issue?
We have checked out the snippet in qlib-main\qlib\contrib\meta\data_selection\model.py and we add self.tn.eval() in validation phase(line 73). Can you specify your question? Where did you find this problem
🐛 Bug Description
To Reproduce
Steps to reproduce the behavior:
Expected Behavior
prediction should not dropout any value or using dynamic batch mean and var for batch normalization. The run_task function should be modified from :
def run_task(self, meta_input, phase): """ A single naive incremental learning task """ ......... with torch.no_grad(): pred = self.framework(meta_input["X_test"].to(self.framework.device), None) return pred.detach().cpu().numpy() to def run_task(self, meta_input, phase): """ A single naive incremental learning task """ ......... with torch.no_grad(): self.framework.eval() pred = self.framework(meta_input["X_test"].to(self.framework.device), None) return pred.detach().cpu().numpy()