neodroidvision.multitask.multi_task_learning_scheduler

Created on 19-09-2021

Functions

cache_backbone_results(model)	Cache and freeze the backbone part of the model and only update heads
switch_target_head(model)	Only single head at a time
common_head_training(model)	Train all heads a the same time