Error in exporting Image classification model

Please provide the following information when requesting support.

• Hardware (T4/V100/Xavier/Nano/etc)
• Network Type (Detectnet_v2/
• TLT Version 3
• Training spec file(If have, please share here)
• How to reproduce the issue ? (This is for errors. Please share the command line and the detailed log here.)

I have been able to train, export and use a couple of Image classifiers (resnet 18/34) using the Image classifier Juniper Notebooks.
This time however I’m stuck during the export Phase.
Training, pruning, retraining and int8 conversion goes well. Once I try exporting the model using !rm -rf $USER_EXPERIMENT_DIR/export/final_model.etlt
!classification export
-m $USER_EXPERIMENT_DIR/output_retrain/weights/resnet_$EPOCH.tlt
-o $USER_EXPERIMENT_DIR/export/final_model.etlt
-k $KEY
–cal_data_file $USER_EXPERIMENT_DIR/export/calibration.tensor
–data_type int8
–batches 10
–cal_cache_file $USER_EXPERIMENT_DIR/export/final_model_int8_cache.bin
-v

the process quits with the following error:

File “/home/vpraveen/.cache/dazel/_dazel_vpraveen/216c8b41e526c3295d3b802489ac2034/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/iva/build_wheel.runfiles/ai_infra/iva/common/export/tensorfile_calibrator.py”, line 79, in get_data_from_source
ValueError: Batch size yielded from data source 8 < requested batch size from calibrator 16

Anyone?

There is no update from you for a period, assuming this is not an issue any more.
Hence we are closing this topic. If need further support, please open a new one.
Thanks

What is your training spec ?
How many images in your training dataset and val dataset?