now the post momentum updates also depend on momentum so we need to add that as an argument
now the post momentum updates also depend on momentum so we need to add that as an argument

snurkabill/pydeeplearn,mihaelacr/pydeeplearn,snurkabill/pydeeplearn,Warvito/pydeeplearn,Warvito/pydeeplearn