diff --git a/examples/scalar_affine.py b/examples/scalar_affine.py index 61bf853..7ee2a6d 100644 --- a/examples/scalar_affine.py +++ b/examples/scalar_affine.py @@ -35,10 +35,11 @@ def main(kappa=0.67, m_sq=-4*0.67, lambd=0.5, n_epochs=1000, batch_size=128, else: hyperparam = dict() - snaps_dir = "../torch-snapshots" - snaps_name= "test_scalar_affine.E200.tar" # if exists resume from here .epoch.tar + snaps_dir = "../torch-snapshots" # folder in which snapshots are stored + # saved snaps MUST BE "..tar" + snaps_name= "test_scalar_affine.E200.tar" # if exists resume training from there snaps_path = os.path.join(snaps_dir, snaps_name) -# snaps_path = '0' # set to 0 if you don't want to save any snapshots +# snaps_path = '0' # set to '0' if you don't want to save any snapshots fit_kwargs = dict( n_epochs=n_epochs, save_every=100,