Method__init__(self, img_size=224, patch_size=16, in_chans=3,
embed_dim=1024, depth=24, num_heads=16,
star/models/multiagent_mae.py:29
Method__init__(self, img_size=224, patch_size=16, in_chans=3,
embed_dim=1024, depth=24, num_heads=16,
star/models/multiagent_mae.py:351
Method__init__(self, img_size=224, patch_size=16, in_chans=3,
embed_dim=1024, depth=24, num_heads=16,
star/models/multiagent_mae.py:688
Method__init__(self, num_embeddings, embedding_dim, commitment_cost, decay, epsilon=1e-5)
star/models/mae_base.py:177
Method__init__(self, img_size=224, patch_size=16, in_chans=3,
embed_dim=1024, depth=24, num_heads=16,
star/models/mae_base.py:260
Methodamortized_complement_unmasking x_ts: [BxAxts, L, D] mask: [BxA, ts, L] -> this is real L ids_restore: [BxA, L] -> this is real L restore, add tempor
star/models/multiagent_mae.py:478
Methodamortized_complement_unmasking x_ts: [BxAxts, L, D] mask: [BxA, ts, L] -> this is real L ids_restore: [BxA, L] -> this is real L restore, add tempor
star/models/multiagent_mae.py:836
Methodamortized_complement_unmasking x_ts: [BxAxts, L, D] mask: [BxA, ts, L] -> this is real L ids_restore: [BxA, L] -> this is real L restore, add tempor
star/models/VQSTAR.py:177
Methodbuild_neighbors_feature_list(self, b, agent_idx, all_warp, num_agent, local_com_mat, device,
size)
star/models/mae_base.py:617
Methodforward(self, imgs1, imgs_next, teacher, trans_matrices, num_agent_tensor, batch_size, mask_ratio=0.7)
star/models/multiagent_mae.py:335
Methodforward_encoder_all x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/multiagent_mae.py:946
Methodforward_encoder_all x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/VQSTAR.py:287
Methodforward_encoder_partial x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/multiagent_mae.py:986
Methodforward_encoder_partial x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/VQSTAR.py:317