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'
|
assert result == 1, 'amp + ddp model failed to complete'
|
||||||
|
|
||||||
# test prediction
|
# test prediction
|
||||||
data = model.test_dataloader
|
data = model.val_dataloader
|
||||||
for batch in data:
|
for batch in data:
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue