Closed Khev closed 5 years ago
1) Enabled code to be run using CPU, achieved using --device 'cpu'. E.g. Full command is
--device 'cpu'
python run_lp.py --dataset=pubmed --dropout=0 --hidden=512 --l2=5e-4 --num_layers=3 --cross_layer=False --patience=500 --device 'cpu'
2) Fixed other minor changes: (i) create 'output' directory if it doesn't exist (ii) added output & tensorboard files to the .gitignore.
1) Enabled code to be run using CPU, achieved using
--device 'cpu'. E.g. Full command ispython run_lp.py --dataset=pubmed --dropout=0 --hidden=512 --l2=5e-4 --num_layers=3 --cross_layer=False --patience=500 --device 'cpu'2) Fixed other minor changes: (i) create 'output' directory if it doesn't exist (ii) added output & tensorboard files to the .gitignore.