latent_model.fit failing

[Shounak Chakraborty]

Hi,

ValueError: Found unexpected keys that do not correspond to any Model output: dict_keys(['out_x', 'out_y']). Expected: ['functional_3', 'functional_9']

b_s = 128

Getting the above error after running :
hist = latent_model.fit(latent_generator(lat_out, lat_out_var, lat_x, lat_x_var, lat_y, lat_y_var, y_traj[:,:out_secs_steps,:1], _, y_traj[:,:out_secs_steps,1:], _, b_s),
batch_size=b_s, epochs=4, shuffle=True, callbacks=[TqdmCallback(verbose=1)], verbose=0, steps_per_epoch=int(lat_out.shape[0]/b_s))