I’m quite new to torch.jit and Apex. I’m getting this error. What does it mean ?
RuntimeError: _register_state_dict_hook is not supported on ScriptModules

Here’s my code
model, optimizer = amp.initialize(model, optimizer,opt_level='O2',
                                          keep_batchnorm_fp32=False,
                                          loss_scale=config.TRAIN.FP16_LOSS_SCALE,
                                          min_loss_scale=128.0)