X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=cnn-svrt.py;h=7fe2db2d569aefc92e904ff21d11cdfec8e04d61;hp=d6c7169e917e5c864decb77975d6d31c52197109;hb=e9f012349010d2a4f5d2ed0869974611fade32f1;hpb=1b7eb64f1a3de3761ff887b4cfbc25a81a60b00e diff --git a/cnn-svrt.py b/cnn-svrt.py index d6c7169..7fe2db2 100755 --- a/cnn-svrt.py +++ b/cnn-svrt.py @@ -35,10 +35,12 @@ import torch import torchvision from torch import optim +from torch import multiprocessing from torch import FloatTensor as Tensor from torch.autograd import Variable from torch import nn from torch.nn import functional as fn + from torchvision import datasets, transforms, utils # SVRT