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
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
