ctheodoris
/

Geneformer

Inference Endpoints

Model card Files Files and versions Community

Update geneformer/emb_extractor.py

#303

by hchen725 - opened Feb 26

base: refs/heads/main

←

from: refs/pr/303

Discussion Files changed

Files changed (1) hide show

geneformer/emb_extractor.py +1 -1

geneformer/emb_extractor.py CHANGED Viewed

@@ -565,7 +565,7 @@ class EmbExtractor:
                 filtered_input_data, cell_state, self.nproc
             )
         downsampled_data = pu.downsample_and_sort(filtered_input_data, self.max_ncells)
-        model = pu.load_model(self.model_type, self.num_classes, model_directory)
         layer_to_quant = pu.quant_layers(model) + self.emb_layer
         embs = get_embs(
             model,

                 filtered_input_data, cell_state, self.nproc
             )
         downsampled_data = pu.downsample_and_sort(filtered_input_data, self.max_ncells)
+        model = pu.load_model(self.model_type, self.num_classes, model_directory, mode = "eval")
         layer_to_quant = pu.quant_layers(model) + self.emb_layer
         embs = get_embs(
             model,