Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "b65830e0c77947a859e5b72b3da9051af3a99b3d"
Commit 4e6edc32 authored by thomwolf's avatar thomwolf
Browse files

hop

parent f55b60b9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/run_classifier.py
View file @ 4e6edc32
......@@ -443,7 +443,7 @@ def main():
nb_eval_steps += 1
if len(preds) == 0:
preds.append(logits.detach().cpu().numpy())
out_label_ids = label_ids.detach().cpu().numpy())
out_label_ids = label_ids.detach().cpu().numpy()
else:
preds[0] = np.append(
preds[0], logits.detach().cpu().numpy(), axis=0)
......@@ -523,7 +523,7 @@ def main():
nb_eval_steps += 1
if len(preds) == 0:
preds.append(logits.detach().cpu().numpy())
out_label_ids = label_ids.detach().cpu().numpy())
out_label_ids = label_ids.detach().cpu().numpy()
else:
preds[0] = np.append(
preds[0], logits.detach().cpu().numpy(), axis=0)
......
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