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

Update unified_cell_classifier.py

Browse files
Files changed (1) hide show
  1. unified_cell_classifier.py +2 -2
unified_cell_classifier.py CHANGED
@@ -130,7 +130,7 @@ class UnifiedCellClassifier(nn.Module):
130
  "4": "Myeloid_cells_classifier",
131
  "5": "NK_cells_classifier",
132
  "7": "TRAV1_2_CD8plus_T_cells_classifier",
133
- "8": "gd_T_cells_classfier"
134
  }
135
 
136
  def _create_classifier_from_config(self, config: Dict):
@@ -270,7 +270,7 @@ class UnifiedCellClassifier(nn.Module):
270
  model.sub_classifiers[sub_name] = model._create_classifier_from_config(sub_config)
271
 
272
  # Carica i pesi del sub-classificatore
273
- sub_weights_path = get_file_path(f"sub_classifiers/{sub_name}.bin")
274
  sub_state_dict = torch.load(sub_weights_path)
275
  model.sub_classifiers[sub_name].load_state_dict(sub_state_dict, strict=False)
276
 
 
130
  "4": "Myeloid_cells_classifier",
131
  "5": "NK_cells_classifier",
132
  "7": "TRAV1_2_CD8plus_T_cells_classifier",
133
+ "8": "gd_T_cells_classifier"
134
  }
135
 
136
  def _create_classifier_from_config(self, config: Dict):
 
270
  model.sub_classifiers[sub_name] = model._create_classifier_from_config(sub_config)
271
 
272
  # Carica i pesi del sub-classificatore
273
+ sub_weights_path = get_file_path(f"sub_classifiers/{sub_name}/{sub_name}.bin")
274
  sub_state_dict = torch.load(sub_weights_path)
275
  model.sub_classifiers[sub_name].load_state_dict(sub_state_dict, strict=False)
276