ivensamdh
/

fp16test

Image Classification

Inference Endpoints

Model card Files Files and versions Community

ivensamdh commited on Jan 30, 2023

Commit

1db6857

•

1 Parent(s): b13a339

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -41,7 +41,7 @@
   ],
   "problem_type": "multi_label_classification",
   "semantic_loss_ignore_index": 255,
-  "torch_dtype": "float32",
   "transformers_version": "4.26.0",
   "use_absolute_position_embeddings": false,
   "use_auxiliary_head": true,

   ],
   "problem_type": "multi_label_classification",
   "semantic_loss_ignore_index": 255,
+  "torch_dtype": "float16",
   "transformers_version": "4.26.0",
   "use_absolute_position_embeddings": false,
   "use_auxiliary_head": true,