Upload folder using huggingface_hub

models/steps_20_batch_200_RNN_4096_relu_rf_012_DoG_True_periodic_False_lr_00001_weight_decay_00001/most_recent_model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8f7050cd3c660329c5c49e1bc935e66cb9eaebeb42caff69930d6ee793c8f6c1
+oid sha256:d8daea843f89ec9adcbf75e675720d9b231e3f38990cdf266844df4dc74b2d39
 size 83954107

trainer.py

@@ -62,7 +62,7 @@ class Trainer(object):
 
         return loss.item(), err.item()
 
-    def train(self, n_epochs: int = 1000, n_steps=10, save=True):
+    def train(self, n_epochs: int = 1, n_steps=10, save=True):
         """
         Train model on simulated trajectories.