leobcc commited on
Commit ·
ab8b8cf
1
Parent(s): 749aa2c
Debugging inncremental sampling
Browse files- code/train.py +1 -1
code/train.py
CHANGED
|
@@ -74,7 +74,7 @@ def main(opt):
|
|
| 74 |
gpus=1,
|
| 75 |
accelerator="gpu",
|
| 76 |
callbacks=[checkpoint_callback],
|
| 77 |
-
max_epochs=opt.epochs_increment_interval+(i+1)*opt.epochs_increment_interval,
|
| 78 |
check_val_every_n_epoch=50,
|
| 79 |
logger=logger,
|
| 80 |
log_every_n_steps=1,
|
|
|
|
| 74 |
gpus=1,
|
| 75 |
accelerator="gpu",
|
| 76 |
callbacks=[checkpoint_callback],
|
| 77 |
+
max_epochs=opt.model.epochs_increment_interval+(i+1)*opt.model.epochs_increment_interval,
|
| 78 |
check_val_every_n_epoch=50,
|
| 79 |
logger=logger,
|
| 80 |
log_every_n_steps=1,
|