问题描述:在AIStudio工作台中,启动环境
按照样例配置好训练集、验证集之后,输入指令
! python train.py \
--dataset_dir "data" \
--device "gpu" \
--max_seq_length 128 \
--model_name "ernie-3.0-medium-zh" \
--batch_size 16 \
--early_stop \
--epochs 1
并运行之后,并没有模型训练更新Loss的日志打出。通过在train.py中打印跟踪信息的方式,定位到此时模型实际上并未启动训练。而两周前我用同样的代码,是能够正常运行的。
解决办法:换一个更高级别的环境,如下图,
即可完美解决。为何这样就能解决也不太清楚,有了解内幕的欢迎留言。