Skip to content

Commit

Permalink
Update conv_tests.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
rbSparky authored Sep 30, 2024
1 parent 2e0a4fb commit ed8e647
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions GNNLux/test/layers/conv_tests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@
l = TransformerConv((6, ein) => 8, heads = 2, gating = true, bias_qkv = true)
test_lux_layer(rng, l, g, x, outputsize = (16,), e = e, container = true)

l = TransformerConv((6, ein) => 8, heads = 2, concat = false, skip_connection = true)
test_lux_layer(rng, l, g, x, outputsize = (16,), e = e, container = true)
# l = TransformerConv((6, ein) => 8, heads = 2, concat = false, skip_connection = true)
# test_lux_layer(rng, l, g, x, outputsize = (16,), e = e, container = true)
end

@testset "GCNConv" begin
Expand Down

0 comments on commit ed8e647

Please sign in to comment.