- You can download the model files (Pytorch versions) and use them just like you would any other saved model. For example, if you save the model to a directory with the name
bert_model, you can load the model like this;
model = ClassificationModel(“bert”, “bert_model”)
Helpful info for downloading models
- The discriminator model is used for any downstream task. The generator is typically only used when training the language model itself.
- I’m not sure I understand this one. Do you mean that you have a set of root_causes and a set of actions, where each root_cause needs to be mapped to te correct action?