"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "648d968cfc69074eaf51df3d337100f9805b030e"
Unverified Commit 56b6a4eb authored by Ramon Zhou's avatar Ramon Zhou Committed by GitHub
Browse files

[GraphBolt] Update GB node classification example learning rate (#7000)

parent 8da604d7
...@@ -287,7 +287,9 @@ def evaluate(args, model, graph, features, itemset, num_classes): ...@@ -287,7 +287,9 @@ def evaluate(args, model, graph, features, itemset, num_classes):
def train(args, graph, features, train_set, valid_set, num_classes, model): def train(args, graph, features, train_set, valid_set, num_classes, model):
optimizer = torch.optim.Adam(model.parameters(), lr=args.lr) optimizer = torch.optim.Adam(
model.parameters(), lr=args.lr, weight_decay=5e-4
)
dataloader = create_dataloader( dataloader = create_dataloader(
graph=graph, graph=graph,
features=features, features=features,
...@@ -343,7 +345,7 @@ def parse_args(): ...@@ -343,7 +345,7 @@ def parse_args():
parser.add_argument( parser.add_argument(
"--lr", "--lr",
type=float, type=float,
default=0.0005, default=1e-3,
help="Learning rate for optimization.", help="Learning rate for optimization.",
) )
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment