TC-ResNet
TC-ResNet copied to clipboard
Generate pb file command is error
correct is
python freeze.py --num_classes 12 --checkpoint_path work/v1/TCResNet8Model-1.0/mfcc_40_3010_0.001_mom_l1/TCResNet8Model-30000 --output_name softmax --preprocess_method no_preprocessing --height 49 --width 40 --channels 1 TCResNet8Model --width_multiplier 1.0
meanwhile ,convert_to_tflite command should be
tflite_convert --graph_def_file=work/v1/TCResNet8Model-1.0/mfcc_40_3010_0.001_mom_l1/TCResNet8Model-30000.pb --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE --output_file=work/v1/TCResNet8Model-1.0/mfcc_40_3010_0.001_mom_l1/TCResNet8Model-30000.tflite --inference_type=FLOAT --inference_input_type=FLOAT --input_arrays=input --output_arrays=softmax --allow_custom_ops
thank you
Thanks a lot