From 53f79cd8cfdb829a705aa510c949bc080a9b003e Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Fri, 29 Oct 2021 09:24:33 +0200 Subject: [PATCH] link --- examples/Project.toml | 3 +-- examples/link_prediction_cora.jl | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/Project.toml b/examples/Project.toml index 3d950f665..b4a89fa64 100644 --- a/examples/Project.toml +++ b/examples/Project.toml @@ -1,10 +1,9 @@ [deps] CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" -DiffEqFlux = "aae7a2af-3d4f-5e19-a356-7da93b79d9d0" -DifferentialEquations = "0c46a032-eb83-5123-abaf-570d42b7fbaa" Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c" GraphNeuralNetworks = "cffab07f-9bc2-4db1-8861-388f63bf7694" Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6" MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458" +MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d" NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d" diff --git a/examples/link_prediction_cora.jl b/examples/link_prediction_cora.jl index e2212463a..d6477fae0 100644 --- a/examples/link_prediction_cora.jl +++ b/examples/link_prediction_cora.jl @@ -23,7 +23,7 @@ Base.@kwdef mutable struct Args epochs = 200 # number of epochs seed = 17 # set seed > 0 for reproducibility usecuda = false # if true use cuda (if available) - nhidden = 128 # dimension of hidden features + nhidden = 64 # dimension of hidden features infotime = 10 # report every `infotime` epochs end