Pytorch建模技巧总结

    # 2 ways to save the net
    torch.save(net1, 'net.pkl')  # save entire net
    torch.save(net1.state_dict(), 'net_params.pkl')   # save only the parameters
    
    

    # copy net1's parameters into net3
    net3.load_state_dict(torch.load('net_params.pkl'))
    prediction = net3(x)
    
    

    # Clip weights of discriminator
    for p in discriminator.parameters():
      p.data.clamp_(-opt.clip_value, opt.clip_value)