Merge branch 'main' of https://huggingface.co/spaces/shivamjadhav/albert_latest_96

classifier/Albert_latest.py

@@ -12,7 +12,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}")