tasknet
tasknet copied to clipboard
sileod
Easy multi-task learning with HuggingFace Datasets and Trainer
Hi! I tried the basic 3-task example from the README file, and the training worked fine. Then I tried to save and load the model: Saving the model worked ok:...
Hello, sorry im quite new to writing issues. I trained a joint token classification and sequence classification model. To save it i used this: `trainer.save_model("multi_task/")` However trying to load the...
The class Adapter expects Z in constructor: ``` class Adapter(transformers.PreTrainedModel): config_class = transformers.PretrainedConfig def __init__(self, config, classifiers=None, Z=None, labels_list=[]): super().__init__(config) self.Z= torch.nn.Embedding(len(config.classifiers_size),config.hidden_size, max_norm=1.0).weight if Z==None else Z self.classifiers=torch.nn.ModuleList( [torch.nn.Linear(config.hidden_size,size) for...
