MCPcopy Create free account

hub / github.com/bbaaii/DreamDiffusion / functions

Functions526 in github.com/bbaaii/DreamDiffusion

⨍ Functions 526◇ Types & classes 106
Methodsample
(self)
code/dc_ldm/modules/distributions/distributions.py:35
Functionscale_module
Scale the parameters of a module and return it.
code/dc_ldm/modules/diffusionmodules/util.py:183
Methodshared_step
(self, batch, **kwargs)
code/dc_ldm/models/diffusion/ddpm.py:1027
Functionsmaller_the_better
(gt, comp)
code/eval_metrics.py:15
Functionssim_metric
(img1, img2)
code/eval_metrics.py:24
Functionstring_begins_with
(prefix, str)
code/dc_ldm/modules/x_transformer.py:102
Functionto_image
(img)
code/eeg_ldm.py:32
Functionto_image
(img)
code/gen_eval_eeg.py:17
Methodtraining_step
(self, batch, batch_idx, optimizer_idx)
code/dc_ldm/models/autoencoder.py:263
Methodtraining_step
(self, batch, batch_idx, optimizer_idx)
code/dc_ldm/models/autoencoder.py:473
Methodtraining_step
(self, batch, batch_idx)
code/dc_ldm/models/diffusion/classifier.py:198
Methodtraining_step
(self, batch, batch_idx)
code/dc_ldm/models/diffusion/ddpm.py:359
Methodtraining_step
(self, batch, batch_idx)
code/dc_ldm/models/diffusion/ddpm.py:1661
Methodunfreeze_cond_stage
(self)
code/dc_ldm/models/diffusion/ddpm.py:672
Methodunfreeze_diffusion_model
(self)
code/dc_ldm/models/diffusion/ddpm.py:664
Methodunfreeze_first_stage
(self)
code/dc_ldm/models/diffusion/ddpm.py:682
Functionuniform_on_device
(r1, r2, shape, device)
code/dc_ldm/models/diffusion/ddpm.py:43
Functionuniq
(arr)
code/dc_ldm/modules/attention.py:15
Functionunpatchify
x: (N, L, patch_size) imgs: (N, 1, num_voxels)
code/sc_mbm/utils.py:115
Methodvalidation_step
(self, batch, batch_idx)
code/dc_ldm/models/autoencoder.py:285
Methodvalidation_step
(self, batch, batch_idx)
code/dc_ldm/models/autoencoder.py:494
Methodvalidation_step
(self, batch, batch_idx)
code/dc_ldm/models/diffusion/classifier.py:210
Methodvalidation_step
(self, batch, batch_idx)
code/dc_ldm/models/diffusion/ddpm.py:467
Methodvalidation_step
(self, batch, batch_idx)
code/dc_ldm/models/diffusion/ddpm.py:1800
Functionwandb_finish
()
code/eeg_ldm.py:29
Functionwandb_init
(config)
code/gen_eval_eeg.py:35
← previous501–526 of 526, ranked by callers