机器学习的小学生

参考： PyTorch在64位Windows下的Conda包： https://zhuanlan.zhihu.com/p/26871672Pytorch官方文档： https://pytorch.org/docs/versions.html

#coding=UTF-8import torch import torch.nn as nnimport mathimport numpy as npfrom PIL import Image,ImageDrawimport matplotlib.pyplot as pltimport collectionsimport matplotlib.cm as cmfrom tor...

#coding=UTF-8import numpy as npimport torchfrom torch.autograd import Variableimport timeheight = 4000width = 3000height2 = 6000width2 = 5000a = np.random.random((height,width))b = np.r...

1. 保存和加载整个模型(1) 保存和加载整个模型torch.save(model_object, 'model.pkl')model = torch.load('model.pkl')(2) 仅保存和加载模型参数(推荐使用)torch.save(model_object.state_dict(), 'params.pkl')model_object.load_stat...

# coding: utf-8from __future__ import print_functionimport copyimport clickimport cv2import numpy as npimport torchfrom torch.autograd import Variablefrom torchvision import models, transfor...

import torchimport torch.nn as nnimport torchvision.transforms as transformsimport torchvision.datasets as dsetsfrom torch.autograd import Variable'''STEP 1: LOADING DATASET'''train_dataset =...