Skip to content

Commit

Permalink
comment
Browse files Browse the repository at this point in the history
  • Loading branch information
vpuri3 committed Sep 10, 2023
1 parent 0e34b5a commit d60abb0
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
1 change: 0 additions & 1 deletion src/GeometryLearning.jl
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ using DocStringExtensions

# PDE stack
using FourierSpaces
using FourierSpaces: linspace

# ML Stack
using Lux
Expand Down
3 changes: 2 additions & 1 deletion src/train.jl
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,8 @@ function train_model(
cb_stats(p, st)
end

# TODO: output a train.log file with timings
# TODO - output a train.log file with timings
# add ProgressMeters.jl

# save statistics
statsfile = open(joinpath(dir, "statistics.txt"), "w")
Expand Down

0 comments on commit d60abb0

Please sign in to comment.