Yuto2007 commited on
Commit
e763d45
·
verified ·
1 Parent(s): 646df12

Update unified_cell_classifier.py

Browse files
Files changed (1) hide show
  1. unified_cell_classifier.py +1 -1
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