Fix imports
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
#python -m intersimple.expert --env=IntersimpleReward --min_timesteps=200 --env_args='{agent:51}' --path='NormalizedIntersimpleExpert_IntersimpleRewardAgent51.pkl'
|
#python -m expert --env=IntersimpleReward --min_timesteps=200 --env_args='{agent:51}' --path='NormalizedIntersimpleExpert_IntersimpleRewardAgent51.pkl'
|
||||||
#python -m intersimple.expert --env=IntersimpleReward --min_timesteps=200 --env_args='{agent:51}' --policy_args='{mu:0.005}' --path='NormalizedIntersimpleExpert_IntersimpleRewardAgent51Mu.005.pkl'
|
#python -m expert --env=IntersimpleReward --min_timesteps=200 --env_args='{agent:51}' --policy_args='{mu:0.005}' --path='NormalizedIntersimpleExpert_IntersimpleRewardAgent51Mu.005.pkl'
|
||||||
#python -m intersimple.expert --env=IntersimpleReward --min_timesteps=200 --env_args='{agent:51}' --policy_args='{mu:0.001}' --path='NormalizedIntersimpleExpert_IntersimpleRewardAgent51Mu.001.pkl'
|
#python -m expert --env=IntersimpleReward --min_timesteps=200 --env_args='{agent:51}' --policy_args='{mu:0.001}' --path='NormalizedIntersimpleExpert_IntersimpleRewardAgent51Mu.001.pkl'
|
||||||
python -m intersimple.expert --env=NRasterized --min_timesteps=200 --env_args='{agent:51,width:36,height:36,m_per_px:2}' --policy_args='{mu:0.001}' --path='NormalizedIntersimpleExpertMu.001_NRasterizedAgent51w36h36mppx2.pkl'
|
python -m expert --env=NRasterized --min_timesteps=200 --env_args='{agent:51,width:36,height:36,m_per_px:2}' --policy_args='{mu:0.001}' --path='NormalizedIntersimpleExpertMu.001_NRasterizedAgent51w36h36mppx2.pkl'
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from imitation.algorithms import adversarial, bc
|
|||||||
from imitation.data import rollout
|
from imitation.data import rollout
|
||||||
from imitation.util import logger
|
from imitation.util import logger
|
||||||
|
|
||||||
from intersimple.intersimple import NRasterized
|
from intersim.envs.intersimple import NRasterized
|
||||||
|
|
||||||
from gail.discriminator import CnnDiscriminator
|
from gail.discriminator import CnnDiscriminator
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
# %%
|
# %%
|
||||||
from stable_baselines3 import PPO
|
from stable_baselines3 import PPO
|
||||||
from intersimple.intersimple import NRasterized, speed_reward
|
from intersim.envs.intersimple import NRasterized, speed_reward
|
||||||
import functools
|
import functools
|
||||||
|
|
||||||
model_name = "ppo_speed_image_lowres"
|
model_name = "ppo_speed_image_lowres"
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
# %%
|
# %%
|
||||||
from stable_baselines3 import PPO
|
from stable_baselines3 import PPO
|
||||||
from intersimple.intersimple import NRasterizedRandomAgent, speed_reward
|
from intersim.envs.intersimple import NRasterizedRandomAgent, speed_reward
|
||||||
import functools
|
import functools
|
||||||
|
|
||||||
model_name = "ppo_speed_image_lowres_random"
|
model_name = "ppo_speed_image_lowres_random"
|
||||||
|
|||||||
Reference in New Issue
Block a user