Fix @epochs as well

darsnack · darsnack · commit 7e1001392bbf · 2022-02-04T09:58:15.000-06:00
diff --git a/src/optimise/train.jl b/src/optimise/train.jl
@@ -1,4 +1,4 @@
-using ProgressLogging: @withprogress, @logprogress
+using ProgressLogging: @progress, @withprogress, @logprogress
 import Zygote: Params, gradient
 
 """

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-using ProgressLogging: @withprogress, @logprogress`
	`1`	`+using ProgressLogging: @progress, @withprogress, @logprogress`
`2`	`2`	`import Zygote: Params, gradient`
`3`	`3`
`4`	`4`	`"""`