MCPcopy Create free account

hub / github.com/cghezhang/MANN / functions

Functions34 in github.com/cghezhang/MANN

⨍ Functions 34◇ Types & classes 5
↓ 3 callersMethodget_NNbias
(self, controlweights, batch_size)
ExpertWeights.py:47
↓ 3 callersMethodget_NNweight
(self, controlweights, batch_size)
ExpertWeights.py:39
↓ 3 callersMethodinitial_bias
(self, shape)
Gating.py:45
↓ 3 callersMethodinitial_weight
(self, shape)
Gating.py:37
↓ 2 callersMethod_apply_sparse_shared
(self, grad, var, indices, scatter_add)
AdamW.py:172
↓ 1 callersMethodbuild_model
(self)
MANN.py:66
↓ 1 callersMethodfp
(self)
Gating.py:50
↓ 1 callersMethodgetParameter
(self, epoch)
AdamWParameter.py:40
↓ 1 callersMethodinitial_alpha
(self)
ExpertWeights.py:32
↓ 1 callersMethodinitial_alpha_np
(self)
ExpertWeights.py:23
↓ 1 callersMethodinitial_beta
(self)
ExpertWeights.py:36
↓ 1 callersMethodlearningRateCosineSGDR
(self, epoch)
AdamWParameter.py:23
↓ 1 callersFunctionmain
()
main.py:17
↓ 1 callersMethodtrain
(self)
MANN.py:122
↓ 1 callersMethodweightDecayNormalized
(self)
AdamWParameter.py:35
FunctionNormalize
(X, axis, savefile= None)
Utils.py:9
Method__init__
rng
Gating.py:8
Method__init__
Construct a new Adam optimizer. Initialization: ``` m_0 <- 0 (Initialize initial 1st moment vector) v_0 <- 0 (Initialize initial 2nd
AdamW.py:38
Method__init__
rng
ExpertWeights.py:9
Method__init__
(self, nEpochs = 200, Te=20, Tmult=2, LR=0.001, weightDecay=0.025, batchSize=32, nBatches = 0)
AdamWParameter.py:7
Method__init__
(self, num_joints, num_styles, rng, sess,
MANN.py:13
Method_apply_dense
(self, grad, var)
AdamW.py:146
Method_apply_sparse
(self, grad, var)
AdamW.py:211
Method_create_slots
(self, var_list)
AdamW.py:118
Method_finish
(self, update_ops, name_scope)
AdamW.py:227
Method_get_beta_accumulators
(self)
AdamW.py:115
Method_prepare
(self)
AdamW.py:139
Method_resource_apply_dense
(self, grad, var)
AdamW.py:159
Method_resource_apply_sparse
(self, grad, var, indices)
AdamW.py:223
Method_resource_scatter_add
(self, x, i, v)
AdamW.py:217
Functionbuild_path
(path)
Utils.py:3
FunctiongetInput
(data, index_joint)
Gating.py:77
Functionsave_EP
(alpha, beta, filename, num_experts)
ExpertWeights.py:56
Functionsave_GT
(weight, bias, filename)
Gating.py:85