Commit 5491cb86 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

debug

parent 506e76fa
......@@ -1531,7 +1531,8 @@ def cross_validation(model, validation_loader, device, validation_shape, tar_ind
cursor = 0
with torch.no_grad():
for batch_idx, (data, target) in enumerate(tqdm.tqdm(validation_loader, desc='validation compute', mininterval=1, disable=None)):
target = target.squeeze()
if target.dim() != 1:
target = target.squeeze()
target = target.to(device)
batch_size = target.shape[0]
data = data.squeeze().to(device)
......
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