Any recommended ways to combine Torchvision pretrained models and MultiTaskModule? #716
-
|
Seems somewhat hard since MultiTaskModule expects task id argument. |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 1 reply
-
|
Hi @dsoselia ! What do you mean with combining? Your model can include a pretrained model inside, so that the forward of your whole model takes task id as argument but you don't forward it to the pretrained submodule. I think it depends on what you have to do here. |
Beta Was this translation helpful? Give feedback.
-
|
Basically, you have to create a wrapper for the original model. If your model is defined as a feature extractor+classifier it's really easy. |
Beta Was this translation helpful? Give feedback.
Hi @dsoselia ! What do you mean with combining? Your model can include a pretrained model inside, so that the forward of your whole model takes task id as argument but you don't forward it to the pretrained submodule. I think it depends on what you have to do here.