basic hack for osx

This commit is contained in:
jkennedyvz 2022-05-22 18:02:03 -07:00
parent 0d19d8ec8e
commit 8aadbfef4a
3 changed files with 6 additions and 5 deletions

View file

@ -112,6 +112,7 @@ class nn():
config.gpu_options.force_gpu_compatible = True
config.gpu_options.allow_growth = True
config.allow_soft_placement = True
nn.tf_sess_config = config
if nn.tf_sess is None: