def METHOD_NAME(self, init_ckpt):
    assert 'model' in init_ckpt
    if self.multi_gpu:
        self.model.module.load_state_dict(init_ckpt['model'])
    else:
        self.model.load_state_dict(init_ckpt['model'])