jittor
/
JGAN
7
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

update cgan

This commit is contained in:
zwy 2020-05-16 15:49:18 +08:00
parent b535892ef8
commit a6b4c1f3fd
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models/cgan/cgan.py
View File

@ -6,7 +6,9 @@ import os
import numpy as np
import math
from jittor import nn
jt.flags.use_cuda = 1
if jt.has_cuda:
jt.flags.use_cuda = 1
parser = argparse.ArgumentParser()
parser.add_argument('--n_epochs', type=int, default=100, help='number of epochs of training')