Skip to content
Snippets Groups Projects
Commit 518c0fa3 authored by Sanghyun Son's avatar Sanghyun Son
Browse files

fix naming error

parent 3d1d9b03
No related branches found
No related tags found
1 merge request!1Jan 09, 2018 updates
......@@ -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()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment