marksgraham / pretrained_generative_models

Pretrained generative models for MONAI Generative

Pretrained generative models

This repository provides some pre-trained models for use with MONAI Generative.

To load use:

model = torch.hub.load("marksgraham/pretrained_generative_models", model='ddpm_2d', verbose=True)