Skip to content

Commit

Permalink
WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
rbSparky committed Aug 2, 2024
1 parent 786b200 commit 01767ee
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions GNNLux/src/layers/conv.jl
Original file line number Diff line number Diff line change
Expand Up @@ -642,16 +642,7 @@ end
σ
end

"""
function NNConv(ch::Pair{Int, Int}, σ = identity;
init_weight = glorot_uniform,
init_bias = zeros32,
use_bias::Bool = true,
add_self_loops::Bool = true,
use_edge_weight::Bool = false,
allow_fast_activation::Bool = true)
"""
# fix args order

function NNConv(ch::Pair{Int, Int}, nn, σ = identity;
aggr = +,
init_bias = zeros32,
Expand Down

0 comments on commit 01767ee

Please sign in to comment.