From 68129af61bdc71d8dee4fe10fc170ef7f9abffa2 Mon Sep 17 00:00:00 2001 From: Francisco Utrera Date: Thu, 5 Aug 2021 17:36:07 -0700 Subject: [PATCH] took out the ch --- chop/utils/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/chop/utils/data.py b/chop/utils/data.py index 0c94e20..7928080 100644 --- a/chop/utils/data.py +++ b/chop/utils/data.py @@ -148,8 +148,8 @@ def __call__(self, img): return img.add(rgb.view(3, 1, 1).expand_as(img)) IMAGENET_PCA = { - 'eigval':ch.Tensor([0.2175, 0.0188, 0.0045]), - 'eigvec':ch.Tensor([ + 'eigval':torch.Tensor([0.2175, 0.0188, 0.0045]), + 'eigvec':torch.Tensor([ [-0.5675, 0.7192, 0.4009], [-0.5808, -0.0045, -0.8140], [-0.5836, -0.6948, 0.4203],