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

posted @ 2024-01-26 16:42  星辰大海初心不变  阅读(3)  评论(0)    收藏  举报