def _setup(self, variant):
set_seed(variant['run_params']['seed'])
self._variant = variant
gpu_options = tf.compat.v1.GPUOptions(allow_growth=True)
self._session = tf.compat.v1.Session(
config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
tf.keras.backend.set_session(self._session)
self.train_generator = None
self._built = False
def _setup(self, variant):
# Set the current working directory such that the local mode
# logs into the correct place. This would not be needed on
# local/cluster mode.
if ray.worker._mode() == ray.worker.LOCAL_MODE:
os.chdir(os.getcwd())
set_seed(variant['run_params']['seed'])
if variant['run_params'].get('run_eagerly', False):
tf.config.experimental_run_functions_eagerly(True)
self._variant = variant
set_gpu_memory_growth(True)
self.train_generator = None
self._built = False
def _setup(self, variant):
# Set the current working directory such that the local mode
# logs into the correct place. This would not be needed on
# local/cluster mode.
if ray.worker._mode() == ray.worker.LOCAL_MODE:
os.chdir(os.getcwd())
set_seed(variant['run_params']['seed'])
self._variant = variant
gpu_options = tf.compat.v1.GPUOptions(allow_growth=True)
self._session = tf.compat.v1.Session(
config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
tf.compat.v1.keras.backend.set_session(self._session)
self.train_generator = None
self._built = False