Types & classes — github.com/KingJamesSong/OrientationDiffusion

CodeHub MCPcopy Create free account

hub / github.com/KingJamesSong/OrientationDiffusion / types & classes

Types & classes25 in github.com/KingJamesSong/OrientationDiffusion

⨍ Functions 151◇ Types & classes 25

↓ 48 callersClassSelfAttention

↓ 44 callersClassDoubleConv

↓ 33 callersClassDown

↓ 33 callersClassUp

↓ 4 callersClassOrientationDiffusion

train_orientation_diffusion.py:89

↓ 4 callersClassUNet

↓ 3 callersClassEMA

↓ 3 callersClassJointDiffusion

train_nse_cond.py:263

↓ 3 callersClassJointDiffusion

train_nse.py:260

↓ 3 callersClassUNet2AmpPhase

↓ 3 callersClassUNet2AmpPhase_Cond

↓ 3 callersClassUNet2PhaseAmp

↓ 3 callersClassUNet2PhaseAmp_Cond

↓ 2 callersClassNavierStokesVelocityDataset

↓ 2 callersClassNavierStokesVorticitySeqDataset

Returns a 3-step vorticity sequence [ω_{t-1}, ω_t, ω_{t+1}] with shape [3, H, W]. Assumes HDF5 structure: f[mode]["vx"], f[mode]["vy"] with s

↓ 1 callersClassBMPDataset

↓ 1 callersClassFIDEvaluation

fid_evaluation.py:22

↓ 1 callersClassUNet_conditional

ClassDoubleConv1x1