Skip to content

AutoMTL

Multi-Task Model (parent class)

AutoMTL

Home
Example
Example
- mobilenetv2
Documentation
Documentation
- Node
- Multi-Task Model (parent class) Multi-Task Model (parent class)
  Table of contents
  - mtl_model
    
    function

MTL Model

mtl_model

mtl_model.mtl_model()

Parent class for customized multi-task model.

function

compute_depth() should be called at the end of the __init__() for the child model class inherited from mtl_model. See example in mobilenetv2.py.
policy_reg() will be called automatically when running alter_train_with_reg() in Trainer.
Note: If choose to not inherit from mtl_model, please run alter_train() in Trainer at the second alter_train stage.