swithmn1
swithmn1
RUNTIME_EXCEPTION : Non-zero status code returned while running Gather node. Name:'Gather.8' Status Message: /onnxruntime_src/onnxruntime/core/providers/common.h:23 int64_t onnxruntime::HandleNegativeAxis(int64_t, int64_t) axis >= -tensor_rank && axis
这个参数下设置dataloader_num_works 和dataloader_pip_memmory ，精度一致上不去，可以在GPU上运行，但是去掉这两个参数 之后，只能在CPU下跑，但是精度慢慢上去，但是训练的很慢
您好，我有135w张数据，在单卡上进行训练，启动训练脚本后，一直卡在 loading data 这里，请教您一下这是什么原因。 数据格式 *.jpg *.txt 放在一个文件夹里，加载数据很慢，有什么提速的办法吗