Spaces:

raja5259
/

EraV2S21_Raj

Sleeping

raja5259 commited on Jun 27

Commit

dee3d77

•

1 Parent(s): c3ea996

update RohanGivenCode.py

Files changed (1) hide show

RohanGivenCode.py CHANGED Viewed

@@ -264,7 +264,7 @@ def train_the_model(train_loader):
 #From here inference
-def infer_the_model(device, test_loader, save1_or_load0, max_length):
     x, y = test_loader.next_batch()
     model = get_model(device)
     if save1_or_load0 == 0:

 #From here inference
+def infer_the_model(device, test_loader, save1_or_load0, max_length, num_return_sequences):
     x, y = test_loader.next_batch()
     model = get_model(device)
     if save1_or_load0 == 0: