removed dummy d
This commit is contained in:
parent
e4313b0b3d
commit
b684fdf502
|
@ -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()
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue