Fixing existing bug (#420)

* fixing 5 types errors for lr

Author: shenyangHuang

examples/linkproppred/tgat.py

@@ -24,7 +24,7 @@
 parser.add_argument('--bsize', type=int, default=200, help='batch size')
 parser.add_argument('--device', type=str, default='cpu', help='torch device')
 parser.add_argument('--epochs', type=int, default=10, help='number of epochs')
-parser.add_argument('--lr', type=str, default=0.0001, help='learning rate')
+parser.add_argument('--lr', type=float, default=0.0001, help='learning rate')
 parser.add_argument('--dropout', type=str, default=0.1, help='dropout rate')
 parser.add_argument('--n-heads', type=int, default=2, help='number of attention heads')
 parser.add_argument(

examples/linkproppred/tgn.py

@@ -33,7 +33,7 @@
 parser.add_argument('--bsize', type=int, default=200, help='batch size')
 parser.add_argument('--device', type=str, default='cpu', help='torch device')
 parser.add_argument('--epochs', type=int, default=30, help='number of epochs')
-parser.add_argument('--lr', type=str, default=0.0001, help='learning rate')
+parser.add_argument('--lr', type=float, default=0.0001, help='learning rate')
 parser.add_argument('--time-dim', type=int, default=100, help='time encoding dimension')
 parser.add_argument('--embed-dim', type=int, default=100, help='attention dimension')
 parser.add_argument('--memory-dim', type=int, default=100, help='memory dimension')

examples/linkproppred/tncn.py

@@ -40,7 +40,7 @@
     choices=[2, 4, 8],
     help='k-th hop common neighbour (CN) embedding extraction (select from 2/4/8)',
 )
-parser.add_argument('--lr', type=str, default=0.0001, help='learning rate')
+parser.add_argument('--lr', type=float, default=0.0001, help='learning rate')
 parser.add_argument('--time-dim', type=int, default=100, help='time encoding dimension')
 parser.add_argument('--embed-dim', type=int, default=100, help='attention dimension')
 parser.add_argument('--memory-dim', type=int, default=100, help='memory dimension')

examples/nodeproppred/tgat.py

@@ -24,7 +24,7 @@
 parser.add_argument('--bsize', type=int, default=200, help='batch size')
 parser.add_argument('--device', type=str, default='cpu', help='torch device')
 parser.add_argument('--epochs', type=int, default=10, help='number of epochs')
-parser.add_argument('--lr', type=str, default=0.0001, help='learning rate')
+parser.add_argument('--lr', type=float, default=0.0001, help='learning rate')
 parser.add_argument('--dropout', type=str, default=0.1, help='dropout rate')
 parser.add_argument('--n-heads', type=int, default=2, help='number of attention heads')
 parser.add_argument(

examples/nodeproppred/tgn.py

@@ -29,7 +29,7 @@
 parser.add_argument('--bsize', type=int, default=200, help='batch size')
 parser.add_argument('--device', type=str, default='cpu', help='torch device')
 parser.add_argument('--epochs', type=int, default=30, help='number of epochs')
-parser.add_argument('--lr', type=str, default=0.0001, help='learning rate')
+parser.add_argument('--lr', type=float, default=0.0001, help='learning rate')
 parser.add_argument('--time-dim', type=int, default=100, help='time encoding dimension')
 parser.add_argument('--embed-dim', type=int, default=100, help='attention dimension')
 parser.add_argument('--memory-dim', type=int, default=100, help='memory dimension')