parent
ddd0f4e747
commit
2c4b1e521e
|
@ -142,7 +142,7 @@ function my_custom_train!(loss, ps, data, opt)
|
||||||
for d in data
|
for d in data
|
||||||
gs = gradient(ps) do
|
gs = gradient(ps) do
|
||||||
training_loss = loss(d...)
|
training_loss = loss(d...)
|
||||||
# Insert what ever code you want here that needs Training loss, e.g. logging
|
# Insert whatever code you want here that needs Training loss, e.g. logging
|
||||||
return training_loss
|
return training_loss
|
||||||
end
|
end
|
||||||
# insert what ever code you want here that needs gradient
|
# insert what ever code you want here that needs gradient
|
||||||
|
|
Loading…
Reference in New Issue