Skip to content
Snippets Groups Projects
Select Git revision
  • f88030af4eb954b094578b35d8704376de73ec76
  • master default protected
2 results

backup.sh

Blame
  • user avatar
    manxilin authored
    2a1b1ca5
    History
    Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    backup.sh 497 B
    # ENZYMES
    python -m train --datadir=data --bmname=ENZYMES --cuda=3 --max-nodes=100 --num-classes=6
    
    # ENZYMES - Diffpool
    python -m train --bmname=ENZYMES --assign-ratio=0.1 --hidden-dim=30 --output-dim=30 --cuda=1 --num-classes=6 --method=soft-assign
    
    # DD
    python -m train --datadir=data --bmname=DD --cuda=0 --max-nodes=500 --epochs=1000 --num-classes=2
    
    # DD - Diffpool
    python -m train --bmname=DD --assign-ratio=0.1 --hidden-dim=64 --output-dim=64 --cuda=1 --num-classes=2 --method=soft-assign