Skip to content

Commit

Permalink
Check w/ tests
Browse files Browse the repository at this point in the history
  • Loading branch information
rbSparky committed Jan 5, 2024
1 parent 0201ea2 commit 3803280
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/layers/conv.jl
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ test_graphs = [g1, g_single_vertex]
y = l(g, x)
@test y[1, 1] w[1] / (d[1] * d[2]) + w[2] / (d[1] * d[3])
@test y[1, 2] w[3] / (d[2] * d[1]) + w[4] / (d[2] * d[3])
@test y l(g, x, w) # checking without custom
@test y l(g, x, w, custom_norm_fn) # checking without custom

# test gradient with respect to edge weights
w = rand(T, 6)
Expand Down

0 comments on commit 3803280

Please sign in to comment.