"torchvision/transforms/_functional_tensor.py" did not exist on "aeafa912585b5398f595a632d49a4e5925644db7"
-
ptrblck authored
* add state_dict, load_state_dict * add test_restoring, test_loss_scale_decrease * disable amp outputs for checkpoint tests * add test for amp.state_dict, cleanup * add state_dict patch, add test * fixed testing, cleanup * add readme for checkpointing * add docs to source/amp * add review changes to doc
dec4fdd6
