Attempting 8 epochs

SujeethJinesh · Nov 1, 2022 · 958cd43 · 958cd43
1 parent 0e712ad
commit 958cd43
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 5 deletions.
diff --git a/__pycache__/main.cpython-39.pyc b/__pycache__/main.cpython-39.pyc
diff --git a/main.py b/main.py
@@ -737,7 +737,7 @@ def on_train_epoch_start(self, trainer, pl_module):
         trainer_kwargs["callbacks"] = [instantiate_from_config(callbacks_cfg[k]) for k in callbacks_cfg]
         trainer_kwargs["max_steps"] = trainer_opt.max_steps
 
-        trainer = Trainer(accelerator="mps", devices=1, max_epochs=3)
+        trainer = Trainer(accelerator="mps", devices=1, max_epochs=8)
         trainer.logdir = logdir  ###
 
         # data

diff --git a/regularization_images/.gitattributes b/regularization_images/.gitattributes
diff --git a/regularization_images/.gitignore b/regularization_images/.gitignore