Skip to content
Snippets Groups Projects
Commit 1c7ca8e6 authored by JGlombitza's avatar JGlombitza
Browse files

delete last changes

parent 65c74795
No related branches found
No related tags found
No related merge requests found
......@@ -9,17 +9,11 @@ from keras.utils import plot_model
import keras.backend.tensorflow_backend as KTF
import utils
import tensorflow as tf
import sys
KTF.set_session(utils.get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial
log_dir="."
try:
CONDOR_ID = os.environ['CONDOR_ID']
except:
sys.exit('Error: Run this script with "pygpu %file"')
# basic trainings parameter
EPOCHS = 10
GRADIENT_PENALTY_WEIGHT = 10
......
......@@ -6,16 +6,11 @@ import mnist
import matplotlib.pyplot as plt
import keras.backend.tensorflow_backend as KTF
from gan import build_generator, build_discriminator, plot_images, make_trainable, get_session
import sys
log_dir="."
KTF.set_session(get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial
try:
CONDOR_ID = os.environ['CONDOR_ID']
except:
sys.exit('Error: Run this script with "pygpu %file"')
# prepare MNIST dataset
data = mnist.load_data()
X_train = data.train_images.reshape(-1, 28, 28, 1) / 255.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment