script for converting retinanet weights from trochvision #2233
Conversation
| *args, | ||
| **kwargs, | ||
| ): | ||
| # TODO: update presets and remove these old config options. They were |
There was a problem hiding this comment.
Have we already updated the preset files? We don't need these shims anymore?
There was a problem hiding this comment.
Not yet, but with this we can update the files, I will do that. Yes we will not the extra params
There was a problem hiding this comment.
Probably good practice to ditch this change from this PR then. And add this change to the upcoming PR that updates the presets.
There was a problem hiding this comment.
I will update the presets as well as script is ready and I already generated presets just the uploading to Kaggle is pending
There was a problem hiding this comment.
Ok! I'll just wait for the preset update on this PR then, either way fine. (but let's avoid removing the compat hack before we add the new presets)
Create a script for converting following torchvison retinanet model checkpoint to keras-hub.
retinanet_resnet50_fpn_cocoretinanet_resnet50_fpn_v2_coco