liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-17 17:34:35 +08:00
Code Activity

Update pytorch_nn

Browse Source
This commit is contained in:
Jactus
2020-11-18 19:17:19 +08:00
parent 58bd2339c0
commit d8414b949a
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -193,7 +193,6 @@ class DNNModelPytorch(Model):
w_val_auto = w_val_auto.cuda() w_val_auto = w_val_auto.cuda()
for step in range(self.max_steps): for step in range(self.max_steps):
self.logger.info(step)
if stop_steps >= self.early_stop_rounds: if stop_steps >= self.early_stop_rounds:
if verbose: if verbose:
self.logger.info("\tearly stop") self.logger.info("\tearly stop")
@@ -201,7 +200,6 @@ class DNNModelPytorch(Model):
loss = AverageMeter() loss = AverageMeter()
self.dnn_model.train() self.dnn_model.train()
self.train_optimizer.zero_grad() self.train_optimizer.zero_grad()
self.logger.info("INIT")
choice = np.random.choice(train_num, self.batch_size) choice = np.random.choice(train_num, self.batch_size)
x_batch_auto = x_train_values[choice] x_batch_auto = x_train_values[choice]