tta_wrapper
tta_wrapper copied to clipboard
inference batch_size
Could you explain why the batch_size must be 1 on inference mode?
Could you explain why the batch_size must be 1 on inference mode?
Because your test set is one by one, you can only predict one result at a time. So batc_size is set to 1.