google-deepmind / deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Long rollout feature

Alex-apostolo opened this issue · comments

I am trying to create a long rollout with 1500 steps on a trained model with cylinder_flow. I found a variable called num_steps in the cfd_eval.py file and changed it to 1500 but it fails at this line scalar_data, traj_data = sess.run([scalar_op, traj_ops]) of run_model.py. I was wondering what changes were made to the code to create the long rollouts as shown with the flag example. Also an argument to specify the rollout_steps would be very useful as a new feature.