Tim M b22c929a64 Corrected additional `.md` files with new image		2 年之前
..
.ipynb_checkpoints	b22c929a64 Corrected additional `.md` files with new image	2 年之前
Pytorch	1f1012c841 Intro-to-GAN-Codebase	3 年之前
Tensorflow	1f1012c841 Intro-to-GAN-Codebase	3 年之前
readme.md	b22c929a64 Corrected additional `.md` files with new image	2 年之前
requirements.txt	81468f712e Update requirements.txt	3 年之前

Introduction to Generative Adversarial Network in TensorFlow and PyTorch

This repository contains code for Introduction to Generative Adversarial Network in TensorFlow and PyTorch blogpost.

Package Dependencies

The repository trains the Generative Adversarial Network in both Pytorch and Tensorflow on Fashion-MNIST dataset. It is tested with:

Cuda-11.1
Cudnn-8.0

The Pytorch and Tensorflow scripts require numpy, tensorflow, torch. To get the versions of these packages you need for the program, use pip: (Make sure pip is upgraded: python3 -m pip install -U pip)

pip3 install -r requirements.txt

Instructions

PyTorch

To train the Generative Adversarial Network with Pytorch, please go into the Pytorch folder and execute either the Python script or the Jupyter Notebook.

TensorFlow

To train the Generative Adversarial Network with TensorFlow, please go into the Tensorflow folder and execute either the Python script or the Jupyter Notebook.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.