未验证 提交 c57cf3ef 编写于 作者: J Jindong Wang 提交者: GitHub

Merge pull request #365 from lw0517/master

upd-clip_model
......@@ -152,9 +152,10 @@ class ClipModel(object):
pred = torch.squeeze(indices)
result = torch.cat([pred.view(-1, 1), label.view(-1, 1)], dim=1)
if res is None:
res = torch.zeros_like(result)
res = torch.cat([res, result], dim=0)
res = res[1:, :].cpu().numpy()
res = result
else:
res = torch.cat([res, result], dim=0)
res = res.cpu().numpy()
acc = np.mean(np.array(res)[:, 0] == np.array(res)[:, 1])
return acc, res
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册