Skip to content

Commit c1a31f5

Browse files
committed
hotfix progressbar bug
1 parent 5f591bd commit c1a31f5

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/wh.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,12 @@ function wh_learn(
2929
inputT = Matrix{Float64}(undef, (size(X, 2), 1))
3030
pred = Matrix{Float64}(undef, (1, size(Y, 2)))
3131
deltaW = Matrix{Float64}(undef, (size(X, 2), size(Y, 2)))
32-
verbose && begin
33-
pb = Progress(size(X, 1) * n_epochs)
32+
if verbose
33+
if isnothing(learn_seq)
34+
pb = Progress(size(X, 1) * n_epochs)
35+
else
36+
pb = Progress(size(learn_seq, 1) * n_epochs)
37+
end
3438
end
3539
for j = 1:n_epochs # 100 epochs
3640
for i in learn_seq # for each events

0 commit comments

Comments
 (0)