  
  # Update layers into Keras custom objects
  for name, obj in inspect.getmembers(sys.modules[__name__]):
      if inspect.isclass(obj) and obj.__module__ == __name__:
          get_custom_objects().update({name: obj})
 
</s> remove for name, obj in inspect.getmembers(sys.modules[__name__]):
 </s> add for _name, obj in inspect.getmembers(sys.modules[__name__]): </s> remove     _test_optimizer(k_optimizers.Adam(), target=0.5)
    _test_optimizer(k_optimizers.Adam(decay=1e-3), target=0.5)
 </s> add     _test_optimizer(k_optimizers.Adam(), target=0.45)
    _test_optimizer(k_optimizers.Adam(decay=1e-3), target=0.45) </s> remove     assert ((_BACKEND == "cpu" and keras.__version__ in ("2.3.0-tf", "2.4.0")) or
 </s> add     assert ((_BACKEND == "cpu" and keras.__version__ in ("2.3.0-tf", "2.4.0",
                                                         "2.6.0", "2.7.0", "2.8.0")) or </s> remove     _test_optimizer(optimizers.AdaBelief(), target=0.5)
 </s> add     _test_optimizer(optimizers.AdaBelief(), target=0.45)
        get_custom_objects().update({_name: obj})
  
  # Update layers into Keras custom objects
  for name, obj in inspect.getmembers(sys.modules[__name__]):
      if inspect.isclass(obj) and obj.__module__ == __name__:
        get_custom_objects().update({_name: obj})
keep keep keep keep replace
 Update code to support Tensorflow versions up to 2.8 (#1213)

* Update maximum tf version in setup + requirements

* - bump max version of tf version in launcher
- standardise tf version check

* update keras get_custom_objects  for tf>2.6

* bugfix: force black text in GUI file dialogs (linux)

* dssim loss - Move to stock tf.ssim function

* Update optimizer imports for compatibility

* fix logging for tf2.8

* Fix GUI graphing for TF2.8

* update tests

* bump requirements.txt versions

* Remove limit on nvidia-ml-py

* Graphing bugfixes
  - Prevent live graph from displaying if data not yet available

* bugfix: Live graph. Collect loss labels correctly

* fix: live graph - swallow inconsistent loss errors

* Bugfix: Prevent live graph from clearing during training

* Fix graphing for AMD 
lib/model/optimizers_tf.py