Fix bugs in test

This commit is contained in:
D-X-Y 2021-05-19 16:38:21 +08:00
parent a46095c2ee
commit b50ad2a522
2 changed files with 5 additions and 1 deletions

View File

@ -88,7 +88,10 @@ At this moment, this project provides the following algorithms and scripts to ru
## Requirements and Preparation
Please install `Python>=3.6` and `PyTorch>=1.5.0`. (You could also run this project in lower versions of Python and PyTorch, but may have bugs).
**First of all**, please use `python setup.py install` to install `xautodl` library.
Please install `Python>=3.6` and `PyTorch>=1.5.0`. (You could use lower versions of Python and PyTorch, but may have bugs).
Some visualization codes may require `opencv`.
CIFAR and ImageNet should be downloaded and extracted into `$TORCH_HOME`.

View File

@ -3,6 +3,7 @@
##################################################
import torch
import torch.nn as nn
import unittest
from xautodl.models.shape_searchs.SoftSelect import ChannelWiseInter