I am training a style transfer model (code is almost identical to a the TF guide here) and for some reason my training is stalling after only 9 training steps (ie it basical
