Jittor warm-up competition

This is for the warm-up competition in 2rd jittor challenge.

Team name: forMsGroves

Introduction

This is an implementation of conditional gan(cgan) based on jittor.

The code is built on the given template.

Installation

• Environment：ubuntu, python3.7, jittor1.3.0, cuda11.1
• install requirements:

  pip install opencv-python==4.1.2.30 numpy==1.18.2 Pillow==6.2.1

• install jittor (cuda) :

  sudo apt install python3.7-dev libomp-dev

python3.7 -m pip install jittor

python3.7 -m jittor.test.test_example

python3.7 -m jittor.test.test_cudnn_op

Pre-traind Models

Please check the discriminator_last.pkl and generator_last.pkl.

Training

python cgan.py