Commit 13e9dab6 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

epoch starts at 1

parent 432b8310
......@@ -83,7 +83,7 @@ if __name__ == "__main__":
x_eval, y_eval = build_eval_batches(sequences, voc2idx, mask_idx, args.gpu)
writer = SummaryWriter(log_dir=args.logdir)
for epoch in range(args.epochs):
for epoch in range(1, args.epochs + 1):
cumloss = 0
for x in train_tensors:
output, loss = model(x)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment