Here is how we can create a resnet50d config and save it: | |
py | |
resnet50d_config = ResnetConfig(block_type="bottleneck", stem_width=32, stem_type="deep", avg_down=True) | |
resnet50d_config.save_pretrained("custom-resnet") | |
This will save a file named config.json inside the folder custom-resnet. |