Spaces:
Runtime error
Runtime error
fixed issue in train bool
Browse files
app.py
CHANGED
@@ -231,9 +231,9 @@ optimizer = optim.SGD(network.parameters(), lr=learning_rate,
|
|
231 |
momentum=momentum)
|
232 |
|
233 |
|
234 |
-
def train_and_test(
|
235 |
|
236 |
-
if
|
237 |
# Train for one epoch and test
|
238 |
train_dataset = MNISTAdversarial_Dataset('./data_mnist',TRAIN_TRANSFORM)
|
239 |
|
|
|
231 |
momentum=momentum)
|
232 |
|
233 |
|
234 |
+
def train_and_test(train_model=True):
|
235 |
|
236 |
+
if train_model:
|
237 |
# Train for one epoch and test
|
238 |
train_dataset = MNISTAdversarial_Dataset('./data_mnist',TRAIN_TRANSFORM)
|
239 |
|