Commit d92ceae9 authored by wangsen's avatar wangsen
Browse files

update readme.md

parent ffa11ddb
...@@ -111,11 +111,8 @@ python install -e . ...@@ -111,11 +111,8 @@ python install -e .
单卡运行 gene classification 单卡运行 gene classification
``` ```
cd geneformer/ cd geneformer/
python train.py python train.py
``` ```
详情可以参考 Geneformer/examples/cell_classification.ipynb 详情可以参考 Geneformer/examples/cell_classification.ipynb
......
...@@ -139,7 +139,7 @@ for organ in organ_list: ...@@ -139,7 +139,7 @@ for organ in organ_list:
logging_steps = round(len(organ_trainset)/geneformer_batch_size/10) logging_steps = round(len(organ_trainset)/geneformer_batch_size/10)
# reload pretrained model # reload pretrained model
model = BertForSequenceClassification.from_pretrained("//Geneformer", model = BertForSequenceClassification.from_pretrained("/path/to/Geneformer",
num_labels=len(organ_label_dict.keys()), num_labels=len(organ_label_dict.keys()),
output_attentions = False, output_attentions = False,
output_hidden_states = False).to("cuda") output_hidden_states = False).to("cuda")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment