Update unified_cell_classifier.py
Browse files
unified_cell_classifier.py
CHANGED
|
@@ -275,7 +275,7 @@ class UnifiedCellClassifier(nn.Module):
|
|
| 275 |
model.sub_classifiers[sub_name].load_state_dict(sub_state_dict, strict=False)
|
| 276 |
|
| 277 |
# Carica le label del sub-classificatore
|
| 278 |
-
sub_labels_path = get_file_path(f"sub_classifiers/{sub_name}_id2label.json")
|
| 279 |
with open(sub_labels_path) as f:
|
| 280 |
model.sub_labels[sub_name] = json.load(f)
|
| 281 |
|
|
|
|
| 275 |
model.sub_classifiers[sub_name].load_state_dict(sub_state_dict, strict=False)
|
| 276 |
|
| 277 |
# Carica le label del sub-classificatore
|
| 278 |
+
sub_labels_path = get_file_path(f"sub_classifiers/{sub_name}/{sub_name}_id2label.json")
|
| 279 |
with open(sub_labels_path) as f:
|
| 280 |
model.sub_labels[sub_name] = json.load(f)
|
| 281 |
|