Unverified Commit d70eade4 authored by TianleZhang's avatar TianleZhang Committed by GitHub
Browse files

bug fix type mismatch (#2830)



int has no attribute of item
Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
parent e83d0a80
...@@ -42,9 +42,9 @@ def get_n_params(model): ...@@ -42,9 +42,9 @@ def get_n_params(model):
return pp return pp
def train(model, G): def train(model, G):
best_val_acc = 0 best_val_acc = torch.tensor(0)
best_test_acc = 0 best_test_acc = torch.tensor(0)
train_step = 0 train_step = torch.tensor(0)
for epoch in np.arange(args.n_epoch) + 1: for epoch in np.arange(args.n_epoch) + 1:
model.train() model.train()
logits = model(G, 'paper') logits = model(G, 'paper')
......
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