From a21777fe44e02efdd6b3a93afef00c5c9a9d6418 Mon Sep 17 00:00:00 2001 From: Chaithya G R Date: Mon, 18 Nov 2024 11:35:15 +0100 Subject: [PATCH] [hotfix] Fix back the num_epochs --- examples/GPU/example_fastMRI_UNet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/GPU/example_fastMRI_UNet.py b/examples/GPU/example_fastMRI_UNet.py index 036914f0..230a5472 100644 --- a/examples/GPU/example_fastMRI_UNet.py +++ b/examples/GPU/example_fastMRI_UNet.py @@ -138,7 +138,7 @@ def plot_state(axs, mri_2D, traj, recon, loss=None, save_name=None): # %% # Start training loop -num_epochs = 2 +num_epochs = 100 optimizer = torch.optim.RAdam(model.parameters(), lr=1e-3) losses = [] # Store the loss values and create an animation image_files = [] # Store the images to create a gif