在看官方教程时,无意中发现别人写的一个脚本,非常简洁. 官方教程地址:http://pytorch.org/tutorials/beginner/data_loading_tutorial.html#sphx-glr-beginner-data-loading-tutorial-py 使用的是dlib自带的特征点检测库,初期用来测试还是不错的 """Create a sample face landmarks dataset. Adapted from dlib/python
for i, (input, target) in enumerate(trainloader): # measure data loading time data_time.update(time.time() - end) input, target = input.cuda(), target.cuda() if i==2: def for_hook(module,input, output): print('output values:',output) handle2 = model.
这是pytorch官方的一个例子 官方教程地址:http://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#sphx-glr-beginner-blitz-cifar10-tutorial-py 代码如下 # coding=utf-8 import torch.nn as nn import torch.nn.functional as F from torch.autograd import Variable import