Closed NickleDave closed 3 years ago
adds ability to specify whether model should be pretrained, and to specify a path to pre-trained weights that should be loaded
TransferTrainer.from_config
Trainer.from_config
TrainConfig
build
state_dict
nn.DataParallel
adds ability to specify whether model should be pretrained, and to specify a path to pre-trained weights that should be loaded
TransferTrainer.from_config
andTrainer.from_config
classmethodsTrainConfig
build
functionsbuild
functions unpack state dict from checkpointstate_dict
key, if not, assume checkpoint itself is just state dictnn.DataParallel