Skip to content

Commit

Permalink
[hotfix] Fix back the num_epochs
Browse files Browse the repository at this point in the history
  • Loading branch information
chaithyagr authored Nov 18, 2024
1 parent 2d05f41 commit a21777f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/GPU/example_fastMRI_UNet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a21777f

Please sign in to comment.