Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Versioning] v0.5 bump #2267

Merged
merged 29 commits into from
Jul 10, 2024
Merged

[Versioning] v0.5 bump #2267

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
dcd825c
init
vmoens Jul 3, 2024
14bcb80
empty
vmoens Jul 5, 2024
5364595
Merge remote-tracking branch 'origin/main' into version-0.5
vmoens Jul 9, 2024
724c41f
amend
vmoens Jul 9, 2024
00bba31
amend
vmoens Jul 9, 2024
4937046
amend
vmoens Jul 9, 2024
86ada40
Merge remote-tracking branch 'origin/main' into version-0.5
vmoens Jul 9, 2024
4cb6162
empty
vmoens Jul 9, 2024
eab40f9
amend
vmoens Jul 9, 2024
0334ca5
amend
vmoens Jul 9, 2024
b077da1
amend
vmoens Jul 9, 2024
312b82e
amend
vmoens Jul 9, 2024
07b9082
amend
vmoens Jul 9, 2024
3610510
amend
vmoens Jul 9, 2024
725e6df
amend
vmoens Jul 9, 2024
bd67b9f
amend
vmoens Jul 9, 2024
0cf1c1a
amend
vmoens Jul 9, 2024
cf6344e
Merge remote-tracking branch 'origin/main' into version-0.5
vmoens Jul 9, 2024
1aec524
amend
vmoens Jul 9, 2024
0704caa
amend
vmoens Jul 9, 2024
1ea60c7
amend
vmoens Jul 9, 2024
52336a1
amend
vmoens Jul 9, 2024
de73d67
amend
vmoens Jul 9, 2024
a8182b8
Merge remote-tracking branch 'origin/version-0.5' into version-0.5
vmoens Jul 9, 2024
9e481a6
amend
vmoens Jul 9, 2024
d9a7178
amend
vmoens Jul 9, 2024
971c2ed
Merge remote-tracking branch 'origin/main' into version-0.5
vmoens Jul 10, 2024
4e0596c
amend
vmoens Jul 10, 2024
4315f0c
amend
vmoens Jul 10, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
amend
  • Loading branch information
@vmoens
vmoens committed Jul 9, 2024
commit 0cf1c1aec26fae8bb14d8f95b0fe6524eb71e9b4
4 changes: 0 additions & 4 deletions sota-implementations/dreamer/config.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,6 @@ env:
horizon: 500
n_parallel_envs: 8
device:
_target_: dreamer_utils._default_device
device: null

collector:
total_frames: 5_000_000
Expand Down Expand Up @@ -39,8 +37,6 @@ optimization:
networks:
exploration_noise: 0.3
device:
_target_: dreamer_utils._default_device
device: null
state_dim: 30
rssm_hidden_dim: 200
hidden_dim: 400
Expand Down
6 changes: 3 additions & 3 deletions sota-implementations/dreamer/dreamer_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ def transform_env(cfg, env):

def make_environments(cfg, parallel_envs=1, logger=None):
"""Make environments for training and evaluation."""
func = functools.partial(_make_env, cfg=cfg, device=cfg.env.device)
func = functools.partial(_make_env, cfg=cfg, device=_default_device(cfg.env.device))
train_env = ParallelEnv(
parallel_envs,
EnvCreator(func),
Expand All @@ -138,7 +138,7 @@ def make_environments(cfg, parallel_envs=1, logger=None):
train_env = transform_env(cfg, train_env)
train_env.set_seed(cfg.env.seed)
func = functools.partial(
_make_env, cfg=cfg, device=cfg.env.device, from_pixels=cfg.logger.video
_make_env, cfg=cfg, device=_default_device(cfg.env.device), from_pixels=cfg.logger.video
)
eval_env = ParallelEnv(
1,
Expand Down Expand Up @@ -332,7 +332,7 @@ def make_collector(cfg, train_env, actor_model_explore):
init_random_frames=cfg.collector.init_random_frames,
frames_per_batch=cfg.collector.frames_per_batch,
total_frames=cfg.collector.total_frames,
policy_device=instantiate(cfg.collector.device),
policy_device=_default_device(cfg.collector.device),
env_device=train_env.device,
storing_device="cpu",
)
Expand Down
Loading