文章目录
一、环境的配置
bash
# zheP04_cmp_AST
$ conda create -n P04_cmp_AST python=3.9
$ conda activate P04_cmp_AST
$ conda install -y anaconda::scikit-learn
$ conda install -y conda-forge::scipy
$ conda install -y conda-forge::numpy
# 根据自己的cuda版本来安装相应的pytorch
$ conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
$ conda install -y conda-forge::tqdm
$ conda install -y conda-forge::efficientnet-pytorch
$ conda install -y conda-forge::tifffile
$ conda install -y conda-forge::pillow
# 我用anaconda装不上matplotlib，所以用了pip
$ pip install -y matplotlib二、预处理
修改源代码中config.py文件的第5行和第6行 （注意这两个目录不要有包含关系，要不然程序会出错）
三、训练
- 运行train_teacher.py文件
- 运行train_student.py文件
- 运行eval.py文件
四、问题
（1）问题一：ValueError: Last value of index pointer should be less than the size of index and data arrays
这是应为scipy的版本太低，按照我的测试，scipy 1.13就可以运行