Update unified_cell_classifier.py
Browse files
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": "
|
| 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 |
|