Function transition

models/conditional.py:40–43

(batch_size, dim=1000)

def transition(batch_size, dim=1000):
    model = Sequential()
    model.add(CondDreamyRNN(unroll=True, output_dim=z_dim, output_length=out_leng, return_sequences=True, activation="tanh", batch_input_shape=(batch_size, time+out_leng, z_dim+2)))
    return model


def get_model(sess, image_shape=(80, 160, 3), gf_dim=64, df_dim=64, batch_size=64,