neuron-testing-cache
/
neuronxcc-2.15.128.0+56dc5a86
/0_REGISTRY
/0.0.25.dev0
/inference
/convnextv2
/hf-internal-testing
/tiny-random-ConvNextV2Model
/2ca247a4de43e9a75997.json
{"architectures": ["ConvNextV2Model"], "depths": [2, 2, 3, 2], "drop_path_rate": 0.0, "hidden_act": "gelu", "hidden_sizes": [10, 20, 30, 40], "id2label": {"0": "LABEL_0", "1": "LABEL_1", "2": "LABEL_2", "3": "LABEL_3", "4": "LABEL_4", "5": "LABEL_5", "6": "LABEL_6", "7": "LABEL_7", "8": "LABEL_8", "9": "LABEL_9"}, "image_size": 224, "initializer_range": 0.02, "label2id": {"LABEL_0": 0, "LABEL_1": 1, "LABEL_2": 2, "LABEL_3": 3, "LABEL_4": 4, "LABEL_5": 5, "LABEL_6": 6, "LABEL_7": 7, "LABEL_8": 8, "LABEL_9": 9}, "layer_norm_eps": 1e-12, "model_type": "convnextv2", "neuron": {"auto_cast": null, "auto_cast_type": null, "compiler_type": "neuronx-cc", "compiler_version": "2.15.128.0+56dc5a86", "disable_fallback": false, "disable_fast_relayout": false, "dynamic_batch_size": true, "inline_weights_to_neff": true, "optlevel": "2", "output_attentions": false, "output_hidden_states": false, "static_batch_size": 1, "static_image_size": 224, "static_num_channels": 3, "static_patch_size": 4}, "num_channels": 3, "num_stages": 4, "out_features": ["stage2", "stage3", "stage4"], "out_indices": [2, 3, 4], "patch_size": 4, "stage_names": ["stem", "stage1", "stage2", "stage3", "stage4"], "task": "image-classification"} |