But if train a single head like "segm" alone or "depth" alone, what should be the return and combine layers for mobilenet+lwrefinenet. Looking for your support.
I also have other issue, when i load my model trained checkpoints, its says some keys are missing, should i also have to load optimizer state_dict?
I am looking at your documentation of your model parameters like return and combine layers for MTLRefinenet+mobilenet v2 . https://github.com/DrSleep/DenseTorch/blob/dev/docs/Models.md You have mentioned as below![image](https://user-images.githubusercontent.com/64223954/220309493-5ef2d0cd-4b41-4065-a07f-498b8d925144.png)
But if train a single head like "segm" alone or "depth" alone, what should be the return and combine layers for mobilenet+lwrefinenet. Looking for your support.
I also have other issue, when i load my model trained checkpoints, its says some keys are missing, should i also have to load optimizer state_dict?