removed dummy d

This commit is contained in:
William Falcon 2019-07-24 10:55:17 -04:00
parent e4313b0b3d
commit b684fdf502
1 changed files with 8 additions and 0 deletions

View File

@ -62,6 +62,14 @@ def main():
# correct result and ok accuracy
assert result == 1, 'amp + ddp model failed to complete'
# test prediction
data = model.test_dataloader
for batch in data:
break
out = model(data[0])
print(out)
clear_tt_dir()