TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

style

Browse Source
This commit is contained in:
Mike J Innes 2018-06-26 15:09:21 +01:00
parent 2723c9ee04
commit 836e3872b6
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/tracker/scalar.jl
View File

@ -9,12 +9,8 @@ tracker(x::TrackedReal) = x.tracker
track(f::Call, x::Real) = TrackedReal(Tracked(f, x, zero(x)))
function back!(x::TrackedReal)
if isinf(x)
error("Loss is Inf")
end
if isnan(x)
error("Loss is NaN")
end
isinf(x) && error("Loss is Inf")
isnan(x) && error("Loss is NaN")
return back!(x, 1)
end