Abdullah-Nazhat
commited on
Update train.py
Browse files
train.py
CHANGED
@@ -166,7 +166,7 @@ def test(dataloader, model, loss_fn):
|
|
166 |
|
167 |
# apply train and test
|
168 |
|
169 |
-
logname = "/PATH/Experiments_cifar10/logs_linearizer/logs_cifar10.csv"
|
170 |
if not os.path.exists(logname):
|
171 |
with open(logname, 'w') as logfile:
|
172 |
logwriter = csv.writer(logfile, delimiter=',')
|
|
|
166 |
|
167 |
# apply train and test
|
168 |
|
169 |
+
logname = "/PATH/Linearizer/Experiments_cifar10/logs_linearizer/logs_cifar10.csv"
|
170 |
if not os.path.exists(logname):
|
171 |
with open(logname, 'w') as logfile:
|
172 |
logwriter = csv.writer(logfile, delimiter=',')
|