My tf skill is quite rough… Here’s the relevant part in improved gan I believe: https://github.com/openai/improved-gan/blob/master/imagenet/build_model.py#L60 9
Maybe it’s here: https://github.com/santi-pdp/segan/blob/master/model.py#L184 11