README.md

metadata

datasets:
  - mnist
language:
  - zh
pipeline_tag: unconditional-image-generation
tags:
  - code

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)