mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-24 12:41:49 +08:00
removed dummy d
This commit is contained in:
1 parent
e4313b0b3d
commit
b684fdf502
1 file changed
+8
@@ -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()
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user