hrnet_w18_small_v1 and hrnet_w18_small_v2 keys are wrong in this achitecture_config dict. They must be hrnet_w18_small_model_v1 and hrnet_w18_small_model_v2. If not, when you try to train that models, the library crashes and gives you a key dictionary error. They must be the same that in architectures_config.json.
hrnet_w18_small_v1 and hrnet_w18_small_v2 keys are wrong in this achitecture_config dict. They must be hrnet_w18_small_model_v1 and hrnet_w18_small_model_v2. If not, when you try to train that models, the library crashes and gives you a key dictionary error. They must be the same that in architectures_config.json.