Commit 2b4be78

committed

[Fix] device error

1 parent 305a910 commit 2b4be78Copy full SHA for 2b4be78

File tree

-2

lines changed

-2

lines changed

Lines changed: 1 addition & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -53,8 +53,7 @@ def main() -> None:`
`53`	`53`	`predicts = []`
`54`	`54`	`targets = []`
`55`	`55`	`for inputs, labels in tqdm(val_dataloader):`
`56`		`- # inputs_all_labels = create_test_data(inputs).to(device)`
`57`		`-`
	`56`	`+ inputs = inputs.to(device)`
`58`	`57`	`predict = model.predict(inputs)`
`59`	`58`
`60`	`59`	`predicts.extend(predict.tolist())`

Comments

(0)