- Ubuntu 16.04
- Python 3.7
- CUDA 11.0
- PyTorch 1.7.1
conda env create --name srml --file environment.yml
conda activate srmlThe file structure should be as follows:
SRML/
├── datasets/
├── common/
├── models/
├── checkpoints/
│ ├── cub/
train.py
test.py
README.md
environment.yml