bitorch.models.model_hub

Functions

convert_dtypes	converts types of the values of dict so that they can be easily compared accross dataframes and csvs.
download_version_table	downloads the newest version table from model hub.
get_matching_row	searches the version table dataframe for a row that matches model kwargs
get_model_path	finds the matching row for model_kwargs in version table and path to model artifact for given configuration
lightning_checkpoint_to_state_dict	converts a pytorch lightning checkpoint to a normal torch state dict
load_from_hub	loads the model that matches the requested model configuration in model_kwargs from the model hub.