文章目录
一、环境的配置
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就可以运行