├── LICENSE
├── README.md
├── env_check.py
├── train.py
├── val.py
├── test.py
├── visual.py
├── data
│ ├── data_temp
│ ├── ds1
│ ├── ds2
│ └── ds3
├── model
│ ├── model1
│ ├── model2
│ └── model3
├── log
│ ├── ds1_model1_test.log
│ ├── ds1_model1_train.log
│ └── ds1_model1_val.log
└── output
├── ds1_model1.out
└── ds1_model2.out
- windows or linux
- cuda, cudnn, pytorch-gpu
- conda:
- create new virtual Python environment
- install required packages
conda create -n venv python=3.10
conda activate venv
pip install torch==2.0.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
Model details.
Process of model experiment.
Visualization of training process & model performance.
Analysis & Future work of experiment results.