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

---
# Model Card for Model ID

<!-- Provide a quick summary of what the model is/does. -->

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]

<!-- Provide the basic links for the model. -->
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)