MAS-PyTorch

MAS-PyTorch copied to clipboard

hi, is the classification_head right in utils/model_class? class classification_head(nn.Module): """ Each task has a seperate classification head which houses the features that are specific to that particular task. These features...

leonzgtee

Could you tell me how to apply Local MAS?

chrisyxue

Issue related to frozen layers

1

Hello, I have a question regarding the freeze layers [here](https://github.com/wannabeOG/MAS-PyTorch/blob/master/utils/mas_utils.py#L343). I notice that the frozen layers are set to require gradient, in contrast, the not frozen layers don't require gradient....

TsuTikgiau

When executing main.py, I only find this The training process on the 0 tasks is completed Testing the model now and it crash WHY?

aynesss