class ModelSet:
    def __init__(self):
        self.heightmap = model()
        self.heightmap.compile(optimizer=keras.optimizers.RMSprop(),
            loss=shore_focused_loss)

      outputs = m(inputs)

      outputs = m.heightmap(inputs)

  m.fit(x=training_data[0], y=training_data[1], batch_size=100, epochs=5000)
  tfjs.converters.save_keras_model(m, 'tfjs_model')

  m.heightmap.fit(x=training_data[0], y=training_data[1], batch_size=100, epochs=5000)
  tfjs.converters.save_keras_model(m.heightmap, 'tfjs_model')

  outputs = tf.reshape(m(inputs), (1024, 2048, 1))

  outputs = tf.reshape(m.heightmap(inputs), (1024, 2048, 1))

  m = model.model()
  m.compile(optimizer=keras.optimizers.RMSprop(),
            loss=model.shore_focused_loss)

  m = model.ModelSet()