Skip to content

Commit 47daf31

Browse files
authored
Update train.py
accessing log_interval from the config.py file
1 parent 06a34a7 commit 47daf31

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def plot_losses(running_train_loss, running_val_loss, train_epoch_loss, val_epoc
116116
optimizer = optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr = lr)
117117
loss_fn = nn.MSELoss()
118118

119-
log_interval = 25
119+
log_interval = cfg.log_interval
120120
epochs = cfg.epochs
121121

122122
###

0 commit comments

Comments
 (0)