我使用的是tqdm库,它没有给我进度条,而是给了我类似这样的输出,它只是告诉我迭代:
251it [01:44, 2.39it/s]
知道为什么代码会这样做吗?我想可能是因为我给它传递了一个发电机,但我又一次使用了过去工作过的发电机。我以前从来没有真正搞乱过tdqm格式。以下是部分源代码:
train_iter = zip(train_x, train_y) #train_x and train_y are just lists of elements
....
def train(train_iter, model, criterion, optimizer):
model.train()
total_loss = 0
for x, y in tqdm(train_iter):
x = x.transpose(0, 1)
y = y.transpose(0, 1)
optimizer.zero_grad()
bloss = model.forward(x, y, criterion)
bloss.backward()
torch.nn.utils.clip_grad_norm(model.parameters(), args.clip)
optimizer.step()
total_loss += bloss.data[0]
return total_loss
tqdm
需要知道将执行多少iter(总数)才能显示进度条。你可以试试这个:
相关问题 更多 >
编程相关推荐