CCPL复现
1.环境配置:
第一次安装的环境是python3.6,按照它本身的requirements进行其他环境的安装。结果运行到进入网络模型的部分就运行的很慢,被卡住。
【分析】以后这种情况,先检查是环境的问题,安装的python、cuda不匹配,重新安装环境。
【改进】感觉该代码的结构与CAST比较类型,所以torch直接按照CAST的requestments.txt进行安装的,可以正常用
具体环境安装步骤如下:
conda create -n CCPL_1 python=3.7
conda activate CCPL_1
pip install -r requirements_CAST.txt (按照CAST的环境主要安装好torch)
pip install tensorboardX
pip install tqdm
2.数据集下载:需要用到的数据集特别大,20-30个G左右,用梯子下太费流量
debug时代码中input提供了一些数据,可以用,不需要下载他们的数据。
3.用自己数据集
train的命令:python train.py --gpu 2 (训练20万个iterators速度特别快,2小时之内就训练完了)
test的命令:python Test_metrics.py