changwh5/BigBiGAN-MNIST-150epoch

Model Card for Model ID

This modelcard aims to be a base template for new models. It has been generated using (https://github.com/huggingface/huggingface_hub/blob/main/src/huggingface_hub/templates/modelcard_template.md?plain=1).

Model Details

Model Sources [optional]

Find the code in https://github.com/RKorzeniowski/BigBiGAN-PyTorch

How to use

def load_model(self,model_path):
    self.model.load_state_dict(torch.load(model_path))

def init_GAN(self):
    print('load the pre-trained model')

    data_set = self.config.ds_name
    if 'CIFAR10' in data_set:
        model_path = 'checkpoint_150.pth'
    elif 'MNIST' in data_set:
        model_path = 'checkpoint_150.pth'

    self.load_model(model_path)