From 518c0fa3e39031999984a3dff6bdaf40eeeef89f Mon Sep 17 00:00:00 2001
From: Sanghyun Son <thstkdgus35@snu.ac.kr>
Date: Wed, 8 May 2019 09:14:35 +0900
Subject: [PATCH] fix naming error
---
src/main.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/main.py b/src/main.py
index 5d3c3ea..dbfac3e 100644
--- a/src/main.py
+++ b/src/main.py
@@ -20,9 +20,9 @@ def main():
else:
if checkpoint.ok:
loader = data.Data(args)
- model = model.Model(args, checkpoint)
- loss = loss.Loss(args, checkpoint) if not args.test_only else None
- t = Trainer(args, loader, model, loss, checkpoint)
+ _model = model.Model(args, checkpoint)
+ _loss = loss.Loss(args, checkpoint) if not args.test_only else None
+ t = Trainer(args, loader, _model, _loss, checkpoint)
while not t.terminate():
t.train()
t.test()
--
GitLab