iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bugs in test

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -88,7 +88,10 @@ At this moment, this project provides the following algorithms and scripts to ru
## Requirements and Preparation ## 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`. Some visualization codes may require `opencv`.
CIFAR and ImageNet should be downloaded and extracted into `$TORCH_HOME`. CIFAR and ImageNet should be downloaded and extracted into `$TORCH_HOME`.

1
tests/test_tas.py
View File

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