MLJFlux interfacing update method

[MarkArdman]

Due to the fact that this model uses a non-standard training method I am fairly certain that the update function from the MLJ interface needs to be implemented in a similar manner to the fit! function as MLJFlux will otherwise use it's own fit! method. However another solution is to overload the aforementioned fit! method and avoid re-implementing fit!and update methods altogether.