for epoch in range(1, N_EPOCH + 1):
// Get srcs and targets from data loader
for i, (srcs, targets) in enumerate(train_loader):
for src, target in zip(srcs, targets):
train_loss = train(src, target)
print("[(%d %d%%) %.4f]" %
(epoch, epoch / N_EPOCH * 100, train_loss))
print(translate(srcs[0]), "\n")
print(translate(), "\n")
After Change
for epoch in range(1, N_EPOCH + 1):
// Get srcs and targets from data loader
for i, (srcs, targets) in enumerate(train_loader):
train_loss = train(srcs[0], targets[0]) // Batch is 1
if i % 100 is 0:
print("[(%d %d%%) %.4f]" %
