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

assert

Browse Source
This commit is contained in:
lucidrains 2025-10-19 09:59:42 -07:00
parent 27ed6d0ba5
commit 55574c054e
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

@ -1883,6 +1883,8 @@ class DynamicsWorldModel(Module):
policy_optim: Optimizer | None = None, policy_optim: Optimizer | None = None,
value_optim: Optimizer | None = None value_optim: Optimizer | None = None
): ):
assert experience.is_batched
latents = experience.latents latents = experience.latents
actions = experience.actions actions = experience.actions
old_log_probs = experience.log_probs old_log_probs = experience.log_probs

2
pyproject.toml
View File

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