ctheodoris hchen725 commited on
Commit
d6c634c
1 Parent(s): 075bd53

Update geneformer/emb_extractor.py (#303)

Browse files

- Update geneformer/emb_extractor.py (d1f49b5321b8f1b9709ddd6b27a577f6acc0ddeb)


Co-authored-by: Han Chen <hchen725@users.noreply.huggingface.co>

Files changed (1) hide show
  1. geneformer/emb_extractor.py +1 -1
geneformer/emb_extractor.py CHANGED
@@ -565,7 +565,7 @@ class EmbExtractor:
565
  filtered_input_data, cell_state, self.nproc
566
  )
567
  downsampled_data = pu.downsample_and_sort(filtered_input_data, self.max_ncells)
568
- model = pu.load_model(self.model_type, self.num_classes, model_directory)
569
  layer_to_quant = pu.quant_layers(model) + self.emb_layer
570
  embs = get_embs(
571
  model,
 
565
  filtered_input_data, cell_state, self.nproc
566
  )
567
  downsampled_data = pu.downsample_and_sort(filtered_input_data, self.max_ncells)
568
+ model = pu.load_model(self.model_type, self.num_classes, model_directory, mode = "eval")
569
  layer_to_quant = pu.quant_layers(model) + self.emb_layer
570
  embs = get_embs(
571
  model,