get_wts.py do not work for yolov4-tiny

[batrlatom]

Hi,
having torch v 1.8 on jetson device, I am unable to get a .wts file.

I think that there is a mismatch between the model it expects and the real file you have provided in your README.

   self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Darknet:
	Missing key(s) in state_dict: "total_ops", "total_params", "module_list.total_ops", "module_list.total_params", "module_list.0.total_ops", "module_list.0.total_params", "module_list.1.total_ops", "module_list.1.total_params", "module_list.2.total_ops", 

etc.
etc.
...

[tjuskyzhang]

please try https://github.com/WongKinYiu/PyTorch_YOLOv4/tree/u3_preview with yolov4-tiny.pt