fix(eval-dataset): torch.torch.set_grad_enabled call (+1 more)

[andrewwhitecdw]

Small fixes in global_recon/eval_dataset.py:

fix: torch.torch.set_grad_enabled call raises AttributeError

Fix: Replace:

torch.torch.set_grad_enabled(False)

with:

torch.set_grad_enabled(False)

fix: evaluation loop only processes first two sequences

Fix: Replace:

for sind, seq_name in enumerate(sequences[:2]):

with:

for sind, seq_name in enumerate(sequences):

Files changed

• global_recon/eval_dataset.py