Skip to content
Snippets Groups Projects
Commit 65c74795 authored by JGlombitza's avatar JGlombitza
Browse files

add sys

parent e0d322ef
No related branches found
No related tags found
No related merge requests found
...@@ -9,6 +9,7 @@ from keras.utils import plot_model ...@@ -9,6 +9,7 @@ from keras.utils import plot_model
import keras.backend.tensorflow_backend as KTF import keras.backend.tensorflow_backend as KTF
import utils import utils
import tensorflow as tf 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 KTF.set_session(utils.get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial
......
...@@ -6,7 +6,7 @@ import mnist ...@@ -6,7 +6,7 @@ import mnist
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import keras.backend.tensorflow_backend as KTF import keras.backend.tensorflow_backend as KTF
from gan import build_generator, build_discriminator, plot_images, make_trainable, get_session from gan import build_generator, build_discriminator, plot_images, make_trainable, get_session
import sys
log_dir="." log_dir="."
KTF.set_session(get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial KTF.set_session(get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment