From d8414b949ae7aba700365543645b513082c8517d Mon Sep 17 00:00:00 2001 From: Jactus Date: Wed, 18 Nov 2020 19:17:19 +0800 Subject: [PATCH] Update pytorch_nn --- qlib/contrib/model/pytorch_nn.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py index 82b7d0950..9bad755b6 100644 --- a/qlib/contrib/model/pytorch_nn.py +++ b/qlib/contrib/model/pytorch_nn.py @@ -193,7 +193,6 @@ class DNNModelPytorch(Model): w_val_auto = w_val_auto.cuda() for step in range(self.max_steps): - self.logger.info(step) if stop_steps >= self.early_stop_rounds: if verbose: self.logger.info("\tearly stop") @@ -201,7 +200,6 @@ class DNNModelPytorch(Model): loss = AverageMeter() self.dnn_model.train() self.train_optimizer.zero_grad() - self.logger.info("INIT") choice = np.random.choice(train_num, self.batch_size) x_batch_auto = x_train_values[choice]