Fix bugs in test
This commit is contained in:
parent
a46095c2ee
commit
b50ad2a522
@ -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`.
|
||||
|
@ -3,6 +3,7 @@
|
||||
##################################################
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import unittest
|
||||
|
||||
from xautodl.models.shape_searchs.SoftSelect import ChannelWiseInter
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user