X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=test-svrt.py;h=5c16069d3ce7ff0bced73eb5d079a9e20a74b1b5;hp=cd98f21daec984c869e5be478b0b009ad4cee058;hb=231c2b2d912d7480af0ce9512b12a909a4fe2a3d;hpb=dfca7e16051d1752db7daed892ecb200237e3bb6 diff --git a/test-svrt.py b/test-svrt.py index cd98f21..5c16069 100755 --- a/test-svrt.py +++ b/test-svrt.py @@ -34,7 +34,7 @@ from torch.nn import functional as fn from torchvision import datasets, transforms, utils -from _ext import svrt +import svrt labels = torch.LongTensor(12).zero_() labels.narrow(0, 0, labels.size(0)//2).fill_(1)