Skip to content
This repository was archived by the owner on Sep 28, 2024. It is now read-only.

Commit 41d5f16

Browse files
foldfelisyuehhua
authored andcommitted
Remove redundant code
1 parent 0705c28 commit 41d5f16

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

example/FlowOverCircle/src/FlowOverCircle.jl

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -54,24 +54,6 @@ function get_dataloader(; ts::AbstractRange = LinRange(100, 11000, 10000),
5454
return loader_train, loader_test
5555
end
5656

57-
function FluxTraining.step!(learner, phase::FluxTraining.TrainingPhase, batch)
58-
xs, ys = batch
59-
FluxTraining.runstep(learner, phase, (; xs = xs, ys = ys)) do handle, state
60-
state.grads = FluxTraining._gradient(learner.optimizer, learner.model, learner.params) do model
61-
state.ŷs = model(state.xs)
62-
handle(FluxTraining.LossBegin())
63-
state.loss = learner.lossfn(state.ŷs, state.ys)
64-
handle(FluxTraining.BackwardBegin())
65-
66-
return state.loss
67-
end
68-
69-
handle(FluxTraining.BackwardEnd())
70-
learner.params, learner.model = FluxTraining._update!(
71-
learner.optimizer, learner.params, learner.model, state.grads)
72-
end
73-
end
74-
7557
function train(; cuda = true, η₀ = 1.0f-3, λ = 1.0f-4, epochs = 50)
7658
if cuda && CUDA.has_cuda()
7759
device = gpu

0 commit comments

Comments
 (0)