search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.23k stars 1.24k forks source link

weights not saved properly #476

Open sushant7878 opened 1 year ago

sushant7878 commented 1 year ago

when I try to run save_model.py it runs without error but all the weights are not saved properly. it says

Total params: 6,062,814 Trainable params: 6,056,606 Non-trainable params: 6,208

when it is saved properly,it should be

Total params: 64,429,405 Trainable params: 64,363,101 Non-trainable params: 66,304

as shown in figure below.

can someone help me with case??

problem-1 problem-2

Sanju0392 commented 7 months ago

i am also facing same issue