fixing flataction discriminator to convert to float beforehand, adding necessary forward calls in expert, adding Fire to video creator from model, and trying full run of options gail with new discrimination model

scratch/etienne/intersimple/render_env_from_model.py

@@ -3,19 +3,31 @@ import stable_baselines3 as sb3
 from intersim.envs.intersimple import NRasterized
 
 
-model_name = 'gail_image_singleagent_nocollision'
-model = sb3.PPO.load(model_name)
+def render_env(model_name='gail_image_multiagent_nocollision', agent=51, environment=NRasterized):
+    """
+    Render a video from an model, agent, and environment
+    Args:
+        model_name (str): name of the model
+        agent (int): agent to start the video from
+        environment (gym.Env): gym environment class to render environment on
+    """
 
-env = NRasterized(stop_on_collision=False, width=36, height=36, m_per_px=2, agent=51)
+    model = sb3.PPO.load(model_name)
 
-obs = env.reset()
-i=0
-while True and i < 600:
-    i+=1
-    action, _states = model.predict(obs)
-    obs, rewards, done, info = env.step(action)
-    env.render(mode='post')
-    if done:
-        break
+    env = environment(stop_on_collision=False, width=36, height=36, m_per_px=2, agent=agent)
 
-env.close(filestr='render/'+model_name)
+    obs = env.reset()
+    i=0
+    while True and i < 600:
+        i+=1
+        action, _states = model.predict(obs)
+        obs, rewards, done, info = env.step(action)
+        env.render(mode='post')
+        if done:
+            break
+
+    env.close(filestr='render/'+model_name+'_agent%i'%(agent)) 
+
+if __name__ == '__main__':
+    import fire
+    fire.Fire(render_env)