4371007e154b7882062cab8696b26d520252971b,test/test_keras.py,KerasTests,test_load_model_custom_objects,#KerasTests#,133

Before Change

            y = np.random.random((1, 3, 3))
            model.train_on_batch(x, y)

            _, fname = tempfile.mkstemp(".h5")
            model.save(fname)

            custom_objects = {
                "TestOptimizer": lambda **kwargs: hvd.DistributedOptimizer(

After Change

            y = np.random.random((1, 3, 3))
            model.train_on_batch(x, y)

            with temppath() as fname:
                model.save(fname)

                custom_objects = {
                    "TestOptimizer": lambda **kwargs: hvd.DistributedOptimizer(
                        TestOptimizer(**kwargs))
                }
                new_model = hvd.load_model(fname, custom_objects=custom_objects)
                new_opt = new_model.optimizer

            self.assertEqual(type(new_opt).__module__, "horovod._keras")
            self.assertEqual(type(new_opt).__name__, "TestOptimizer")
            self.assertEqual(K.get_value(opt.lr), K.get_value(new_opt.lr))
            self._check_optimizer_weights(opt, new_opt)

Instances