Add total(f, model) to replace implicit sum(f, Flux.params(model))

Add total(f, model) to replace implicit sum(f, Flux.params(model)) #581