Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/arnavkj1995/WEAVER
/ functions
Functions
311 in github.com/arnavkj1995/WEAVER
⨍
Functions
311
◇
Types & classes
32
Method
forward
( self, obs: dict[str, torch.Tensor], actions: torch.Tensor, tasks: torch.Tens
weaver/wm/model.py:206
Method
forward
Forward pass of FlowWM. Args: obs: Dictionary with patch features per image key (B, T, N, D), and 'stat
weaver/wm/model.py:738
Method
forward
( self, obs: torch.Tensor, actions: torch.Tensor, # t: torch.Tensor, t
weaver/wm/model.py:2052
Method
generate_videos_full_memory
Autoregressive video generation with correct t_memory-spaced memory. obs and actions must cover the full trajectory from frame 0 onward (as
weaver/wm/model.py:1715
Method
get_frame
Return buffer frame at abs_pos; clamp to frame 0 if out of range.
weaver/wm/model.py:1806
Function
handler
(signum, frame)
weaver/robot/panda.py:24
Method
latents
(self)
weaver/datasets/droid.py:177
Function
load_videos_split_top_bottom
Load videos saved by wm_eval.py: all views in one file. Layout: top half = real (views side-by-side), bottom half = pred (same). Width =
weaver/utils/tools.py:282
Function
sample_and_merge_batch
( exp_dataloader, pi_dataloader )
weaver/utils/eval.py:267
Function
save_on_preempt
(signum, frame)
weaver/finetune.py:300
Function
save_on_preempt
(signum, frame)
weaver/reflow.py:630
← previous
301–311 of 311, ranked by callers