[Dev] Added transform argument

create_submission.py

@@ -23,7 +23,7 @@ import torch
 # Local imports
 import bindataset as dataset
 
-def create_submission(model):
+def create_submission(model, transform):
     step_days = 10
     batch_size = 1024
     # We make chunks of num_days consecutive samples; As our dummy predictor

main.py

@@ -79,7 +79,7 @@ if __name__ == "__main__":
         val_loss = test.test(model, valid_loader, f_loss, device)
         print(" Validation : Loss : {:.4f}".format(val_loss))
 
-    create_submission.create_submission(model)
+    create_submission.create_submission(model, None)
     """
     logdir = generate_unique_logpath(top_logdir, "linear")
     print("Logging to {}".format(logdir))