Are you the author? Claim this model
GPU deploy isn't supported for this model type.
1from torch import load 2from models.gan.model import Generator 3 4# Load model 5model = Generator(latent_dim=100, img_channels=1) 6model.load_state_dict(load("generator.pt")) 7model.eval()
