Serialization - added model weights

combo/modules/model.py

@@ -500,6 +500,7 @@ class Model(Module, FromParameters):
         vocabulary_dir = os.path.join(self.serialization_dir, 'vocabulary')
         self.vocab._serialization_dir = vocabulary_dir
         weights_path = os.path.join(self.serialization_dir, 'weights.pth')
+        torch.save(self.state_dict(), weights_path)
         serialized_model = super().serialize()
         serialized_model['weights'] = weights_path
         return serialized_model