Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
raw
history blame contribute delete
288 Bytes
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.