CUDA_VISIBLE_DEVICES=0 python train_landmarks_generator.py --pre_audio_root ./Dataset/wf_audio --landmarks_root ./Dataset/wf_landmarks

[xiaomaofei]

0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]
0it [00:00, ?it/s]。这是对的吗

[xiaomaofei]

降低batch_size =2和batch_size_val =2。原因是我的显卡内存不足

[xiaomaofei]

放弃吧。这个项目不是很好