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)
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
HF Inference API was unable to determine this model's library.