From 717ada526363c641df83b92987e2288d6dedb679 Mon Sep 17 00:00:00 2001 From: heavenlxj Date: Fri, 17 Jun 2022 10:48:07 +0800 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 1b8ac49..d3cde25 100644 --- a/README.md +++ b/README.md @@ -65,7 +65,7 @@ python -u finetune_Drop.py --data ogbg-molhiv --gpu 3 --dropout 0.1 --lr 0.005 - 1. Search Architecture ``` -python model_search.py --gpu 0 --num_layers 5 --epochs 5 --data ogbg-molpcba +python train_search.py --gpu 0 --num_layers 5 --epochs 5 --data ogbg-molpcba --remove_pooling True ``` 2. Finetune the model. @@ -79,7 +79,7 @@ python finetune.py --gpu 0 --dropout 0.5 --lr 0.001 --batch_size 100 --num_layer 1. Search Architecture ``` -python model_search.py --gpu 0 --num_layers 3 --epochs 5 --data ogbg-ppa +python train_search.py --gpu 0 --num_layers 3 --epochs 5 --data ogbg-ppa --remove_pooling True ``` 2. Finetune the model. @@ -92,7 +92,7 @@ python finetune.py --gpu 1 --dropout 0.5 --lr 0.0005 --batch_size 24 --num_layer 1. Search Architecture ``` -python model_search.py --gpu 0 --num_layers 5 --epochs 10 --batch_size 24 --hidden_size 64 --data ogbg-ppa +python train_search.py --gpu 0 --num_layers 5 --epochs 10 --batch_size 24 --hidden_size 64 --data ogbg-ppa --remove_pooling True ``` 2. Finetune the model.