hongshaorou/dreamer4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

evolutionary policy optimization on dreams will be interesting

Browse Source
This commit is contained in:
lucidrains 2025-10-28 10:17:01 -07:00
parent c0450359f3
commit 7acaa764f6
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dreamer4/dreamer4.py
View File

@ -2543,6 +2543,7 @@ class DynamicsWorldModel(Module):
batch_size = 1, batch_size = 1,
agent_index = 0, agent_index = 0,
tasks: int | Tensor | None = None, tasks: int | Tensor | None = None,
latent_gene_ids = None,
image_height = None, image_height = None,
image_width = None, image_width = None,
return_decoded_video = None, return_decoded_video = None,
@ -2658,6 +2659,7 @@ class DynamicsWorldModel(Module):
step_sizes = step_size, step_sizes = step_size,
rewards = decoded_rewards, rewards = decoded_rewards,
tasks = tasks, tasks = tasks,
latent_gene_ids = latent_gene_ids,
discrete_actions = decoded_discrete_actions, discrete_actions = decoded_discrete_actions,
continuous_actions = decoded_continuous_actions, continuous_actions = decoded_continuous_actions,
proprio = noised_proprio_with_context, proprio = noised_proprio_with_context,

2
pyproject.toml
View File

@ -1,6 +1,6 @@
[project] [project]
name = "dreamer4" name = "dreamer4"
version = "0.0.89" version = "0.0.90"
description = "Dreamer 4" description = "Dreamer 4"
authors = [ authors = [
{ name = "Phil Wang", email = "lucidrains@gmail.com" } { name = "Phil Wang", email = "lucidrains@gmail.com" }