huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Larger batches

Browse Source
This commit is contained in:
ebuehrle
2021-11-04 11:01:06 +01:00
parent 361c1ea9d3
commit 0990eb808a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scratch/etienne/intersimple/gail_options_image_random_location.py
View File

@@ -25,13 +25,13 @@ ALL_OPTIONS = [(v,t) for v in [0,2,4,6,8] for t in [5, 10, 20]] # option 0 is sa
def train( def train(
expert_data, expert_data,
expert_batch_size=1024, expert_batch_size=4069,
discriminator_updates_per_round=10, discriminator_updates_per_round=10,
generator_steps=256, generator_steps=256,
generator_total_steps=1024, generator_total_steps=2048,
generator_updates_per_round=10, generator_updates_per_round=10,
discount=0.99, discount=0.99,
epochs=200, epochs=100,
): ):
env = NRasterizedRouteSpeedRandomAgentLocation(**env_settings) env = NRasterizedRouteSpeedRandomAgentLocation(**env_settings)