ton/IronpenGPU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
ton
2023-08-15 19:43:30 +07:00
parent 1775684e38
commit 6b2bfa4628
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/forward.jl
View File

@@ -126,7 +126,8 @@ function (kfn::kfn_1)(input::AbstractArray)
kfn.on_recSignal, kfn.on_recSignal,
kfn.on_subscription, kfn.on_subscription,
) )
# get on_zt4d to on_zt
kfn.on_zt .= reduce(max, kfn.on_zt4d, dims=(1,2))
logit = reshape(kfn.on_zt, (size(input, 1), :)) logit = reshape(kfn.on_zt, (size(input, 1), :))
# error("DEBUG -> kfn forward") # error("DEBUG -> kfn forward")