Index

A

ActNorm (class in nn.modules.actNorm) add() (nn.trainFlowParallel.TMGLowPredictionItem method) AffineCouplingBlock (class in nn.modules.flowLSTMBlock)

AffineCouplingLayer (class in nn.modules.flowAffine) allreduce() (in module utils.parallel) args module

B

BackwardStepConfig (class in args) BackwardStepLoader (class in utils.dataLoader) batch_size (args.BackwardStepConfig attribute) (args.CylinderConfig attribute)

bcolors (class in utils.log) beta (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) BOLD (utils.log.bcolors attribute)

C

calcDivergence() (pc.physicsConstrained.PhysConstrainedLES method) calcNormalizingParams() (utils.dataLoader.TMGLowDataLoader method) calcPressurePoisson() (pc.physicsConstrained.PhysConstrainedLES method) calcVDiv() (nn.trainFlowParallel.TMGLowLoss method) calcVPres() (nn.trainFlowParallel.TMGLowLoss method) CheckerSqueeze (class in nn.modules.flowUtils) clear() (nn.trainFlowParallel.TMGLowPredictionItem method) concat() (nn.trainFlowParallel.TMGLowPredictionItem method)

Conv2dZeros (class in nn.modules.flowUtils) ConvLSTMCell (class in nn.modules.convLSTM) createTestingLoader() (utils.dataLoader.BackwardStepLoader method) (utils.dataLoader.CylinderArrayLoader method) createTrainingLoader() (utils.dataLoader.BackwardStepLoader method) (utils.dataLoader.CylinderArrayLoader method) CylinderArrayLoader (class in utils.dataLoader) CylinderConfig (class in args)

D

DataLoaderAuto (class in utils.dataLoader) DataParallelCriterion (class in utils.parallel) DenseBlock (class in nn.modules.denseBlock)

dx (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) dy (args.BackwardStepConfig attribute) (args.CylinderConfig attribute)

E

Encoder (class in nn.tmGlow) enconding_transition() (nn.tmGlow.Encoder method)

ENDC (utils.log.bcolors attribute) error() (utils.log.Log method)

F

FAIL (utils.log.bcolors attribute) first_encoding() (nn.tmGlow.Encoder method) forward() (nn.modules.actNorm.ActNorm method) (nn.modules.convLSTM.ConvLSTMCell method) (nn.modules.convLSTM.ResidLSTMBlock method) (nn.modules.flowAffine.AffineCouplingLayer method) (nn.modules.flowAffine.LSTMAffineCouplingLayer method) (nn.modules.flowLSTMBlock.AffineCouplingBlock method) (nn.modules.flowLSTMBlock.LSTMCouplingBlock method) (nn.modules.flowLSTMBlock.LSTMFLowBlock method) (nn.modules.flowLSTMBlock.UnNormedAffineCouplingBlock method) (nn.modules.flowUtils.CheckerSqueeze method) (nn.modules.flowUtils.Conv2dZeros method) (nn.modules.flowUtils.LatentEncoder method) (nn.modules.flowUtils.Split method) (nn.modules.flowUtils.Squeeze method) (nn.modules.glowConv.InvertibleConv1x1 method) (nn.modules.glowConv.InvertibleConv1x1LU method) (nn.modules.misc.UpsamplingLinear method) (nn.tmGlow.Encoder method) (nn.tmGlow.LSTMCFlowDecoder method) (nn.tmGlow.TMGlow method) (nn.trainFlowParallel.TMGLowLoss method) (utils.parallel.DataParallelCriterion method)

G

GaussianDiag (class in nn.modules.flowUtils) getGpuMemoryMap() (in module utils.utils) getOutputs() (nn.trainFlowParallel.TMGLowPredictionItem method) getTargets() (nn.trainFlowParallel.TMGLowPredictionItem method)

glow_upscale (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) Grad1Filter2d (class in pc.grad1Filter) Grad2Filter2d (class in pc.grad2Filter)

H

HEADER (utils.log.bcolors attribute)

I

info() (utils.log.Log method) init_data_loaders() (utils.dataLoader.DataLoaderAuto class method) init_hidden() (nn.modules.convLSTM.ConvLSTMCell method)

initLSTMStates() (nn.tmGlow.TMGlow method) inv_weight() (nn.modules.glowConv.InvertibleConv1x1LU method) InvertibleConv1x1 (class in nn.modules.glowConv) InvertibleConv1x1LU (class in nn.modules.glowConv)

L

laplaceGrad() (pc.grad2Filter.Grad2Filter2d method) LatentEncoder (class in nn.modules.flowUtils) likelihood() (nn.modules.flowUtils.GaussianDiag method) loadConfig() (args.Parser method) loadWorkspace() (in module utils.utils) Log (class in utils.log) log() (utils.log.Log method) Log2PI (nn.modules.flowUtils.GaussianDiag attribute)

log_determinant() (nn.modules.glowConv.InvertibleConv1x1 method) log_prob() (nn.modules.flowUtils.GaussianDiag method) logLoss() (utils.log.Log method) logp (nn.trainFlowParallel.TMGLowPredictionItem attribute) logTest() (utils.log.Log method) LSTMAffineCouplingLayer (class in nn.modules.flowAffine) LSTMCFlowDecoder (class in nn.tmGlow) LSTMCouplingBlock (class in nn.modules.flowLSTMBlock) LSTMFLowBlock (class in nn.modules.flowLSTMBlock)

M

main module max_grad_norm (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) mkdirs() (args.Parser method) modelPred() (in module utils.utils) module args main nn.modules.actNorm nn.modules.convLSTM nn.modules.denseBlock nn.modules.flowAffine nn.modules.flowLSTMBlock nn.modules.flowUtils nn.modules.glowConv nn.modules.misc nn.tmGlow nn.trainFlowParallel pc.grad1Filter pc.grad2Filter pc.physicsConstrained utils.dataLoader utils.log utils.parallel utils.utils utils.viz

moment (utils.log.uid attribute)

N

n_gpu (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) nic (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) nn.modules.actNorm module nn.modules.convLSTM module nn.modules.denseBlock module nn.modules.flowAffine module nn.modules.flowLSTMBlock module nn.modules.flowUtils module nn.modules.glowConv module

nn.modules.misc module nn.tmGlow module nn.trainFlowParallel module noise_std (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) NoNormDenseBlock (class in nn.modules.denseBlock) normalizeInputData() (utils.dataLoader.TMGLowDataLoader method) normalizeTargetData() (utils.dataLoader.TMGLowDataLoader method) ntest (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) ntrain (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) nu (args.BackwardStepConfig attribute) (args.CylinderConfig attribute)

O

OKBLUE (utils.log.bcolors attribute)

OKGREEN (utils.log.bcolors attribute)

P

parallel (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) parse() (args.Parser method) Parser (class in args) pc.grad1Filter module pc.grad2Filter module

pc.physicsConstrained module PhysConstrainedLES (class in pc.physicsConstrained) plotNumericalPred() (in module utils.viz) plotVelocityPred() (in module utils.viz) print_progress() (utils.log.Log method)

R

readFluidData() (utils.dataLoader.TMGLowDataLoader method) reconstruct() (nn.tmGlow.TMGlow method) ResidLSTMBlock (class in nn.modules.convLSTM) reverse() (nn.modules.actNorm.ActNorm method) (nn.modules.flowAffine.AffineCouplingLayer method) (nn.modules.flowAffine.LSTMAffineCouplingLayer method) (nn.modules.flowLSTMBlock.AffineCouplingBlock method) (nn.modules.flowLSTMBlock.LSTMCouplingBlock method) (nn.modules.flowLSTMBlock.LSTMFLowBlock method) (nn.modules.flowLSTMBlock.UnNormedAffineCouplingBlock method) (nn.modules.flowUtils.CheckerSqueeze method) (nn.modules.flowUtils.Split method) (nn.modules.flowUtils.Squeeze method) (nn.modules.glowConv.InvertibleConv1x1 method) (nn.modules.glowConv.InvertibleConv1x1LU method) (nn.tmGlow.LSTMCFlowDecoder method)

S

sample() (nn.modules.flowUtils.GaussianDiag method) (nn.tmGlow.TMGlow method) saveWorkspace() (in module utils.utils) setNormalizingParams() (utils.dataLoader.TMGLowDataLoader method)

setupBackwardStepLoaders() (utils.dataLoader.DataLoaderAuto class method) setupCylinderLoaders() (utils.dataLoader.DataLoaderAuto class method) Split (class in nn.modules.flowUtils) Squeeze (class in nn.modules.flowUtils) success() (utils.log.Log method)

T

tback (nn.trainFlowParallel.TMGLowPredictionItem attribute) test() (nn.trainFlowParallel.TrainFlow method) test_batch_size (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) testing_data_dir (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) TMGlow (class in nn.tmGlow) TMGLowDataLoader (class in utils.dataLoader) TMGLowLoss (class in nn.trainFlowParallel) TMGLowPredictionItem (class in nn.trainFlowParallel) toNumpy() (in module utils.utils) toTuple() (in module utils.utils) TrainFlow (class in nn.trainFlowParallel) training (nn.modules.actNorm.ActNorm attribute) (nn.modules.convLSTM.ConvLSTMCell attribute) (nn.modules.convLSTM.ResidLSTMBlock attribute) (nn.modules.denseBlock.DenseBlock attribute) (nn.modules.denseBlock.NoNormDenseBlock attribute) (nn.modules.flowAffine.AffineCouplingLayer attribute) (nn.modules.flowAffine.LSTMAffineCouplingLayer attribute) (nn.modules.flowLSTMBlock.AffineCouplingBlock attribute) (nn.modules.flowLSTMBlock.LSTMCouplingBlock attribute) (nn.modules.flowLSTMBlock.LSTMFLowBlock attribute) (nn.modules.flowLSTMBlock.UnNormedAffineCouplingBlock attribute) (nn.modules.flowUtils.CheckerSqueeze attribute) (nn.modules.flowUtils.Conv2dZeros attribute) (nn.modules.flowUtils.LatentEncoder attribute) (nn.modules.flowUtils.Split attribute) (nn.modules.flowUtils.Squeeze attribute) (nn.modules.glowConv.InvertibleConv1x1 attribute) (nn.modules.glowConv.InvertibleConv1x1LU attribute) (nn.modules.misc.UpsamplingLinear attribute) (nn.tmGlow.Encoder attribute) (nn.tmGlow.LSTMCFlowDecoder attribute) (nn.tmGlow.TMGlow attribute) (nn.trainFlowParallel.TMGLowLoss attribute) (pc.grad1Filter.Grad1Filter2d attribute) (pc.grad2Filter.Grad2Filter2d attribute) (pc.physicsConstrained.PhysConstrainedLES attribute) (utils.parallel.DataParallelCriterion attribute)

training_data_dir (args.BackwardStepConfig attribute) (args.CylinderConfig attribute) TrainingDataset (class in utils.dataLoader) trainParallel() (nn.trainFlowParallel.TrainFlow method) transferNormalizingParams() (utils.dataLoader.TMGLowDataLoader method)

U

uid (class in utils.log) UNDERLINE (utils.log.bcolors attribute) UnNormedAffineCouplingBlock (class in nn.modules.flowLSTMBlock) unsqueeze() (nn.trainFlowParallel.TMGLowPredictionItem method) UpsamplingLinear (class in nn.modules.misc) utils.dataLoader module

utils.log module utils.parallel module utils.utils module utils.viz module

W

WARNING (utils.log.bcolors attribute)

warning() (utils.log.Log method) weight() (nn.modules.glowConv.InvertibleConv1x1LU method)

X

xGrad() (pc.grad1Filter.Grad1Filter2d method) (pc.grad2Filter.Grad2Filter2d method)

Y

yGrad() (pc.grad1Filter.Grad1Filter2d method) (pc.grad2Filter.Grad2Filter2d method)

yPred (nn.trainFlowParallel.TMGLowPredictionItem attribute) yTarget (nn.trainFlowParallel.TMGLowPredictionItem attribute)