removed dummy d
This commit is contained in:
parent
8e131f9d79
commit
5606fd86df
|
@ -63,7 +63,7 @@ def main():
|
|||
assert result == 1, 'amp + ddp model failed to complete'
|
||||
|
||||
# test prediction
|
||||
data = model.test_dataloader
|
||||
data = model.val_dataloader
|
||||
for batch in data:
|
||||
break
|
||||
|
||||
|
|
Loading…
Reference in New Issue