Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
aurorarossi committed Aug 18, 2024
1 parent e509abf commit e4b6f80
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/layers/temporalconv.jl
Original file line number Diff line number Diff line change
Expand Up @@ -35,8 +35,8 @@ function TGCNCell(ch::Pair{Int, Int};
end

function (tgcn::TGCNCell)(h, g::GNNGraph, x::AbstractArray)
= l.conv(g, x)
h, x̃ = l.gru(h, x̃)
= tgcn.conv(g, x)
h, x̃ = tgcn.gru(h, x̃)
return h, x̃
end

Expand Down

0 comments on commit e4b6f80

Please sign in to comment.