Method__init__(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, start_idx=0)
preprocessing/torch_utils/misc.py:123
Method__init__(
self,
*,
ch=128,
out_ch=3,
ch_mult=(1, 1, 2, 2, 4),
num_res_
models/invae.py:280
Functionconst_like(ref, value, shape=None, dtype=None, device=None, memory_format=None)
preprocessing/torch_utils/misc.py:55
Methodforward t: (B, Hh, N, D_rot), D_rot == self.rot_dim rope_ids: None | (N,) | (B,N), indexing flattened HW positions for the
models/pos_embed.py:111