1a1308d62f
add fixes for reproducibility only try to start env if it is closed revision fix normalization and data type Improve README Improve README Tests are passing, Eval pretrained model works, Add gif Update gif Update gif Update gif Update gif Update README Update README update minor Update README.md Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> Update README.md Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> Address suggestions Update thumbnail + stats Update thumbnail + stats Update README.md Co-authored-by: Alexander Soare <alexander.soare159@gmail.com> Add more comments Add test_examples.py
28 lines
382 B
YAML
28 lines
382 B
YAML
# @package _global_
|
|
|
|
eval_episodes: 50
|
|
eval_freq: 7500
|
|
save_freq: 75000
|
|
log_freq: 250
|
|
# TODO: same as simxarm, need to adjust
|
|
offline_steps: 25000
|
|
online_steps: 25000
|
|
|
|
fps: 10
|
|
|
|
dataset_id: pusht
|
|
|
|
env:
|
|
name: pusht
|
|
task: pusht
|
|
from_pixels: True
|
|
pixels_only: False
|
|
image_size: 96
|
|
action_repeat: 1
|
|
episode_length: 300
|
|
fps: ${fps}
|
|
|
|
policy:
|
|
state_dim: 2
|
|
action_dim: 2
|