search

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.2k stars 1.36k forks source link

AttributeError: module 'apex.amp' has no attribute 'state_dict' #1710

Open caoren-shuai opened 11 months ago

caoren-shuai commented 11 months ago

Describe the Bug

Minimal Steps/Code to Reproduce the Bug

**Expected Behavior**

Environment

caoren-shuai commented 11 months ago

import torch from apex import amp print(torch.version) print(torch.version.cuda) print(torch.cuda.amp) print(amp.state_dict)

Traceback (most recent call last): File "/home/cc/PycharmProjects/ffb6d/test.py", line 6, in print(amp.state_dict) AttributeError: module 'apex.amp' has no attribute 'state_dict'

caoren-shuai commented 11 months ago

please help me，thanks ery much