(flags, model, optimizer)
| 82 | |
| 83 | |
| 84 | def build_pretrained_weights(flags, model, optimizer): |
| 85 | # 是否加载之前训练的模型 |
| 86 | pretrain_weights = flags.Global.pretrain_weights |
| 87 | to_use_device = flags.Global.device |
| 88 | if pretrain_weights and os.path.exists(pretrain_weights): |
| 89 | model, _resumed_optimizer, global_state = load_checkpoint(model, pretrain_weights, to_use_device, optimizer) |
| 90 | if flags.Global.resumed_optimizer and _resumed_optimizer is not None: |
| 91 | optimizer = _resumed_optimizer |
| 92 | else: |
| 93 | global_state = {} |
| 94 | model.apply(weight_init) |
| 95 | return model, optimizer, global_state |
| 96 | |
| 97 | |
| 98 | def build_device(flags): |
no test coverage detected