關於訓練數據的問題(About train dataset size problem)
Shaiken opened this issue · comments
您好,
若是訓練數據大小的問題,我該怎麼調整,除了去config調整訓練影像高寬大小和crop影像高寬大小
直接改會出現以下相關問題
hello, if training dataset size problem, how edit config .
Original stack trace for 'Reshape':
File "tools/train_model.py", line 332, in <module>
train_model(args.dataset_dir, weights_path=args.weights_path)
File "tools/train_model.py", line 171, in train_model
train_label_tensor_scale, train_net_output_tensor_scale, max_val=255),
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/ops/image_ops_impl.py", line 3370, in ssim
...
要在產dataset前改大小