Skip to content

Commit

Permalink
updat readme
Browse files Browse the repository at this point in the history
  • Loading branch information
wang-chen committed Jan 29, 2021
1 parent 6f8c627 commit c9b39f5
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,14 @@

For feature graph network (FGN):

python nonlifelong.py --model FGN --optim SGD --lr 1e-3 --save saves/nonlifelong-fgn
python lifelong.py --model FGN --optim SGD --lr 1e-3 --memory-size 5000 --save saves/lifelong-fgn
python nonlifelong.py --model FGN --optim SGD
python lifelong.py --model FGN --optim SGD

For graph attention network (GAT):

python nonlifelong.py --model GAT --optim Adam --lr 1e-3 --save saves/nonlifelong-gat
python lifelong.py --model FGN --optim Adam --lr 1e-3 --memory-size 5000 --save saves/lifelong-gat
python nonlifelong.py --model GAT --optim Adam
python lifelong.py --model FGN --optim Adam

You can also specify the dataset location to be downloaded (Default: /data/datasets). For example:

python nonlifelong.py --data-root ./ --model FGN --optim SGD --lr 1e-3 --save saves/nonlifelong-fgn
python nonlifelong.py --data-root ./ --model FGN --optim SGD

0 comments on commit c9b39f5

Please sign in to comment.