You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I've been working all day trying to figure out how to export qat model for different batch size. So from what I understand, I just need to override dataset_params.val_dataloader_params.batch_size so that I can use different batch size to export the onnx model as defined in sg_trainer.py#L2250.
After a while, I know that it uses the same batch_size as the train_dataloader_params.batch_size after applying pre_launch_callback function because the val_dataloader_params uses/copy train_dataloader_params in pre_launch_callbacks.py#L397
Is there any particular reason why you override it with train_dataloader_params?
The text was updated successfully, but these errors were encountered:
I've been working all day trying to figure out how to export qat model for different batch size. So from what I understand, I just need to override
dataset_params.val_dataloader_params.batch_size
so that I can use different batch size to export the onnx model as defined in sg_trainer.py#L2250.After a while, I know that it uses the same
batch_size
as thetrain_dataloader_params.batch_size
after applyingpre_launch_callback
function because theval_dataloader_params
uses/copytrain_dataloader_params
in pre_launch_callbacks.py#L397Is there any particular reason why you override it with
train_dataloader_params
?The text was updated successfully, but these errors were encountered: