MCPcopy Create free account

hub / github.com/dome272/Diffusion-Models-pytorch / functions

Functions37 in github.com/dome272/Diffusion-Models-pytorch

↓ 6 callersMethod__init__
(self, in_channels, out_channels, emb_dim=256)
modules.py:103
↓ 3 callersFunctionget_data
(args)
utils.py:24
↓ 3 callersFunctionsave_images
(images, path, **kwargs)
utils.py:17
↓ 2 callersMethodnoise_images
(self, x, t)
ddpm.py:30
↓ 2 callersMethodsample
(self, model, n, labels, cfg_scale=3)
ddpm_conditional.py:41
↓ 2 callersFunctionsetup_logging
(run_name)
utils.py:36
↓ 1 callersFunctionlaunch
()
ddpm_conditional.py:112
↓ 1 callersFunctionlaunch
()
ddpm.py:94
↓ 1 callersMethodnoise_images
(self, x, t)
ddpm_conditional.py:32
↓ 1 callersFunctionplot_images
(images)
utils.py:9
↓ 1 callersMethodpos_encoding
(self, t, channels)
modules.py:153
↓ 1 callersMethodpos_encoding
(self, t, channels)
modules.py:217
↓ 1 callersMethodprepare_noise_schedule
(self)
ddpm_conditional.py:29
↓ 1 callersMethodprepare_noise_schedule
(self)
ddpm.py:27
↓ 1 callersMethodreset_parameters
(self, ema_model, model)
modules.py:31
↓ 1 callersMethodsample
(self, model, n)
ddpm.py:39
↓ 1 callersMethodsample_timesteps
(self, n)
ddpm_conditional.py:38
↓ 1 callersMethodsample_timesteps
(self, n)
ddpm.py:36
↓ 1 callersMethodstep_ema
(self, ema_model, model, step_start_ema=2000)
modules.py:23
↓ 1 callersFunctiontrain
(args)
ddpm_conditional.py:66
↓ 1 callersFunctiontrain
(args)
ddpm.py:61
↓ 1 callersMethodupdate_average
(self, old, new)
modules.py:18
↓ 1 callersMethodupdate_model_average
(self, ma_model, current_model)
modules.py:13
Method__init__
(self, noise_steps=1000, beta_start=1e-4, beta_end=0.02, img_size=256, device="cuda")
ddpm_conditional.py:17
Method__init__
(self, noise_steps=1000, beta_start=1e-4, beta_end=0.02, img_size=256, device="cuda")
ddpm.py:16
Method__init__
(self, beta)
modules.py:8
Method__init__
(self, channels, size)
modules.py:36
Method__init__
(self, in_channels, out_channels, mid_channels=None, residual=False)
modules.py:59
Method__init__
(self, in_channels, out_channels, emb_dim=256)
modules.py:80
Method__init__
(self, c_in=3, c_out=3, time_dim=256, device="cuda")
modules.py:129
Method__init__
(self, c_in=3, c_out=3, time_dim=256, num_classes=None, device="cuda")
modules.py:190
Methodforward
(self, x)
modules.py:49
Methodforward
(self, x)
modules.py:72
Methodforward
(self, x, t)
modules.py:96
Methodforward
(self, x, skip_x, t)
modules.py:120
Methodforward
(self, x, t)
modules.py:163
Methodforward
(self, x, t, y)
modules.py:227