removed dummy d
This commit is contained in:
parent
b684fdf502
commit
eb4b3a5752
|
@ -66,7 +66,7 @@ def main():
|
|||
data = model.test_dataloader
|
||||
for batch in data:
|
||||
break
|
||||
out = model(data[0])
|
||||
out = model(batch[0])
|
||||
print(out)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue