Skip to content

Commit

Permalink
add guidance for selection of the hyperparameters
Browse files Browse the repository at this point in the history
  • Loading branch information
TencentAILabHealthcare committed May 23, 2022
1 parent 11f07f2 commit be43114
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,23 @@ python --data_path "test_data_path" --model_path "finetuned_model_path" predict.
```
python -m torch.distributed.launch --data_path "pre-train_data_path" pretrain.py
```
- Guidance for hyperparameter selection

You can select the hyperparameters of the Performer encoder based on your data and task in:
```
model = PerformerLM(
num_tokens = 7,
dim = 200,
depth = 6,
heads = 10
)
```
Hyperparameter|Description | Default | Arbitrary range
--------------|---------------------------------------| ------- | ----------------
num_tokens |Number of bins in expression embedding | 7 | [5, 7, 9]
dim |Size of scBERT embedding vector | 200 | [100, 200]
heads |Number of attention heads of Performer | 10 | [8, 10, 20]
depth |Number of Performer encoder layers | 6 | [4, 6, 8]

# Disclaimer
This tool is for research purpose and not approved for clinical use.
Expand Down

0 comments on commit be43114

Please sign in to comment.