Method__init__(self, dim, dim_out, *, time_emb_dim=None, groups=8, conv_2d=False)
burgers/ddpm_burgers/unet.py:151
Method__init__(
self,
dim,
init_dim = None,
out_dim = None,
dim_mults=(1, 2, 4, 8),
burgers/ddpm_burgers/unet.py:268
Method__init__(
self,
dim,
init_dim = None,
out_dim = None,
dim_mults=(1, 2, 4, 8),
burgers/ddpm_burgers/unet.py:419
Method__init__(self, dataset, batch_size=1, shuffle=True, pin_memory=True, num_workers=1)
burgers/ddpm_burgers/data_burgers_1d.py:244
Method__init__(self, field_sequence_model,
display=None, depth=0, batch=0,
figure_builder=
smoke/phi/viz/dash_gui.py:15
Method__init__(
self,
dim,
init_dim = None,
out_dim = None,
dim_mults=(1, 2, 4, 8),
smoke/ddpm/diffusion_2d.py:257
Method__init__(self, dataset, batch_size=1, shuffle=True, pin_memory=True, num_workers=1)
smoke/ddpm/data_2d.py:225