configuration_bbsnet.py

from typing import List

from transformers import PretrainedConfig

"""
The configuration of a model is an object that 
will contain all the necessary information to build the model.

The three important things to remember when writing you own configuration are the following:

- you have to inherit from PretrainedConfig,
- the __init__ of your PretrainedConfig must accept any kwargs,
- those kwargs need to be passed to the superclass __init__.
"""


class BBSNetConfig(PretrainedConfig):

    """
    Defining a model_type for your configuration is not mandatory,
    unless you want to register your model with the auto classes."""

    model_type = "bbsnet"

    def __init__(self, **kwargs):
        super().__init__(**kwargs)


if __name__ == "__main__":
    """
    With this done, you can easily create and save your configuration like
    you would do with any other model config of the library.
    Here is how we can create a resnet50d config and save it:
    """
    bbsnet_config = BBSNetConfig()
    bbsnet_config.save_pretrained("custom-bbsnet")

    """
    This will save a file named config.json inside the folder custom-resnet. 
    You can then reload your config with the from_pretrained method:
    """
    bbsnet_config = BBSNetConfig.from_pretrained("custom-bbsnet")

    """
    You can also use any other method of the PretrainedConfig class, 
    like push_to_hub() to directly upload your config to the Hub.
    """