Skip to content

Latest commit

 

History

History

gan_mnist_pytorch

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
const.yaml
const.yaml
 
 
data.py
data.py
 
 
distributed.yaml
distributed.yaml
 
 
model_def.py
model_def.py
 
 

README.md

PyTorch MNIST GAN Example

This example demonstrates how to build a simple GAN on the MNIST dataset using Determined's PyTorch API. This example is adapted from this PyTorch Lightning GAN example.

Files

  • model_def.py: The core code for the model. This includes building and compiling the model.
  • data.py: The data loading and preparation code for the model.

Configuration Files

  • const.yaml: Train the model with constant hyperparameter values.
  • distributed.yaml: Same as const.yaml, but instead uses multiple GPUs (distributed training).

To Run

Installation instructions can be found under docs/install-admin.html or at Determined installation page. After configuring the settings in const.yaml, run the following command: det -m <master host:port> experiment create -f const.yaml .