Skip to content

Commit

Permalink
rolled back
Browse files Browse the repository at this point in the history
  • Loading branch information
vganapati committed Jul 6, 2023
1 parent 8acd19a commit ce3514c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def get_args():
parser.add_argument('--epochs', type=int, action='store', dest='epochs',
help='number of epochs', default = 1)
parser.add_argument('--2d', action='store_true', dest='two_d',
help='simulation in 2d as opposed to 3d', default = True)
help='simulation in 2d as opposed to 3d')
parser.add_argument('--lr', type=float, action='store', dest='learning_rate',
help='learning rate', default = 1e-3)
parser.add_argument('-j', type=float, action='store', dest='jitter',
Expand Down

0 comments on commit ce3514c

Please sign in to comment.