Skip to content

Commit 89eec44

Browse files
committed
fix adapt for Network
1 parent 7902670 commit 89eec44

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/NetworkDynamicsCUDAExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ function Adapt.adapt_structure(to, n::Network)
3939
extmap = adapt(to, n.extmap)
4040

4141
Network{exT,gT,typeof(layer),typeof(vb),typeof(mm),eltype(caches),typeof(gbp),typeof(extmap)}(
42-
vb, layer, n.im, caches, mm, gbp, extmap)
42+
vb, layer, n.im, caches, mm, gbp, extmap, getfield(n, :jac_prototype))
4343
end
4444

4545
Adapt.@adapt_structure NetworkLayer

0 commit comments

Comments
 (0)