Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
wengong-jin authored Feb 6, 2022
1 parent ee8c3ce commit 6d183da
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ python baseline_train.py --cdr_type 3 --train_path data/sabdab/hcdr3_cluster/tra
```
The above training script usually consumes 4GB GPU memory. You can also train our AR-GNN baseline by setting `--architecture AR-GNN`.


## Antigen-binding antibody design (Section 4.2)
To train a RefineGNN for this task, please run
```
Expand Down

0 comments on commit 6d183da

Please sign in to comment.