Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/KingJamesSong/OrientationDiffusion
/ functions
Functions
151 in github.com/KingJamesSong/OrientationDiffusion
⨍
Functions
151
◇
Types & classes
25
Method
__init__
(self, in_channels, out_channels, emb_dim=256)
modules.py:169
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:192
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:251
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:316
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:376
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, num_classes=None, device="cuda")
modules.py:442
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:507
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:574
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:641
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, img_size=32, device="cuda")
modules.py:708
Method
__init__
( self, batch_size, dl, sampler, channels=3, model= None,
fid_evaluation.py:23
Method
__init__
(self, noise_steps=100, noise_start=1e-4, noise_end=0.1, coupling_start=3e-5, coupling_end=0.03, img_size=32,
train_orientation_diffusion.py:90
Method
__init__
(self, noise_steps=100, noise_start=1e-4, noise_end=0.1, coupling_start=3e-3, coupling_end=0.045, beta_start=1
train_nse.py:261
Method
__len__
(self)
nse_pde.py:44
Method
__len__
(self)
nse_pde.py:109
Method
__len__
(self)
dataset.py:50
Method
__len__
(self)
dataset.py:82
Method
amplitude_prefactor_local
(self, r, M=4)
train_nse_cond.py:307
Method
amplitude_prefactor_local
(self, r, M=4)
train_nse.py:304
Method
coupling_noise
(self, x)
train_nse_cond.py:282
Method
coupling_noise
(self, x)
train_nse.py:279
Method
coupling_noise_local
(self, x, M=5)
train_orientation_diffusion.py:115
Method
forward
(self, x)
modules.py:49
Method
forward
(self, x)
modules.py:72
Method
forward
(self, x)
modules.py:92
Method
forward
(self, x, t)
modules.py:116
Method
forward
(self, x, t)
modules.py:137
Method
forward
(self, x, skip_x, t)
modules.py:161
Method
forward
(self, x, skip_x, t)
modules.py:185
Method
forward
(self, x, t)
modules.py:226
Method
forward
(self, x, t)
modules.py:288
Method
forward
(self, x, t)
modules.py:350
Method
forward
(self, x, t)
modules.py:413
Method
forward
(self, x, t, y)
modules.py:479
Method
forward
(self, x, phase_coherence, t)
modules.py:544
Method
forward
(self, x, mag, t)
modules.py:611
Method
forward
(self, x, phase_coherence, t, x_cond, phase_cond)
modules.py:678
Method
forward
(self, x, mag, t, cond, mag_cond)
modules.py:745
Function
kl_div_wg
(mu_q, mu_p, sigma_q, K=3, num_samples=10)
train_nse_cond.py:105
Function
kl_div_wg
(mu_q, mu_p, sigma_q, K=3, num_samples=10)
train_nse.py:101
Function
plot_images
(images)
utils.py:22
Function
safe_open_dataset
(path)
nse_pde.py:155
Method
sample_noise
(self, model, step, size, loc, concentration)
train_orientation_diffusion.py:137
Function
save_circular_density_plot
Save a density visualization of data points mapped to a unit circle. Args: data (torch.Tensor): 1D tensor containing angles in rad
utils.py:89
Function
save_one_dim
Save a 1D scatter plot of the data. Args: data (torch.Tensor): 1D tensor containing values to plot. path (str): Path to sav
utils.py:60
Function
save_two_dim
(data, path, range=20)
utils.py:47
Function
spectrum_metrics_batch
Compute slope difference and Wasserstein distance between spectra of real and fake velocity fields. Args: real, fake: torch.Tensor o
train_nse_cond.py:215
Method
unnormalize
(self, data)
dataset.py:72
Function
validate
(args)
train_orientation_diffusion.py:249
Function
validate_fid
(args)
train_orientation_diffusion.py:301
Function
validate_noise
(args)
train_orientation_diffusion.py:287
← previous
101–151 of 151, ranked by callers