Pytorch 加载 模型

ubuntu docker with pytorch, tensorflow

Foundations of Deep Learning | This repo collects 7 of programming labs exercises for Foundations of Deep Learning. In this course, you will learn the basic concepts of deep learning, including the basic principles of neural networks, the basic principles of TensorFlow, Keras and PyTorch, and the...

Example pytorch image

Jupyter Notebook with Pytorch and CUDA GPUs supports

1.保存模型与加载模型 2.冻结一部分参数，训练另一部分参数 3.采用不同的学习率进行训练 1.保存模型与加载 简单的保存模型与加载方法： # 保存整个网络torch.save(net, PATH) # 保存网络中的参数, 速度快，占空间少torch.save(net.state_dict(),PA