Changes for submission-strategyA-reinit

run.py

@@ -282,7 +282,7 @@ class ZEWDPCBaseRun:
                 num_epochs = max(epochs_affordable, num_epochs)
                 print(f"last training epoch is {num_epochs} instead!")
             """start training"""
-            #self.model = load_model()
+            self.model = load_model()
             tr_dset = NPDataset(images=image_lst, labels=label_lst, transform=self.train_transform)
             dataloader = DataLoader(dataset=tr_dset, batch_size=BATCHSIZE, shuffle=True, num_workers=4)
             self.train_model(dataloader, nepoch=num_epochs)