Update classifier/Albert_latest.py

classifier/Albert_latest.py

@@ -11,7 +11,7 @@ class Model:
         checkpoint = torch.load(model_weights, map_location=self.device)
         
         # Load the model's state dictionary
-        self.model.load_state_dict(checkpoint['model_state_dict'])
+        self.model.load_state_dict(checkpoint['model_state_dict'],strict=False)
         self.currepoch = checkpoint['epoch']
         self.loss = checkpoint['loss']
         print(f"Loaded model state: Current epoch {self.currepoch}, current loss {self.loss}")