TypeError: expected sequence object with len >= 0 or a single integer
在GPU训练的代码,预测时也要在GPU
原命令python predict.py换成CUDA_VISIBLE_DEVICES=0 python predict.py
TypeError: expected sequence object with len >= 0 or a single integer
在GPU训练的代码,预测时也要在GPU
原命令python predict.py换成CUDA_VISIBLE_DEVICES=0 python predict.py