def get_generator(latent_dim, input_dim, ngf): # Initialise generator G = Gene(latent_dim, output_dim = input_dim, ngf = ngf) G.apply(weights_init) G = G.to(device) return G