定义了一个类BaseNet并实例化该类:
net=BaseNet()
保存net时报错object has no attribute 'state_dict'
torch.save(net.state_dict(), models_dir)
原因是定义类的时候不是继承nn.Module类,比如:
class BaseNet(object):def __init__(self):
把类定义改为
class BaseNet(nn.Module):def __init__(self):super(BaseNet, self).__init__()