MCPcopy Create free account

hub / github.com/MethodsOfMachineLearning/cabs / functions

Functions23 in github.com/MethodsOfMachineLearning/cabs

↓ 5 callersFunctionbias_variable
(shape, val=0.05)
examples/models/cifar10_2conv_3dense.py:14
↓ 5 callersFunctionweight_variable
(shape, stddev=1e-2)
examples/models/cifar10_2conv_3dense.py:10
↓ 4 callersFunctionbias_variable
(shape)
examples/models/mnist_2conv_2dense.py:12
↓ 4 callersFunctionweight_variable
(shape)
examples/models/mnist_2conv_2dense.py:8
↓ 2 callersFunction_generate_image_and_label_batch
Construct a queued batch of images and labels. Args: image: 3-D Tensor of [height, width, 3] of type.float32. label: 1-D Tensor of type.int3
examples/cifar10_adaptive_batchsize.py:117
↓ 2 callersFunctionconv2d
(x, W)
examples/models/mnist_2conv_2dense.py:16
↓ 2 callersFunctionconv2d
(x, W)
examples/models/cifar10_2conv_3dense.py:18
↓ 2 callersFunctionmax_pool_2x2
(x)
examples/models/mnist_2conv_2dense.py:19
↓ 2 callersFunctionmax_pool_3x3
(x)
examples/models/cifar10_2conv_3dense.py:21
↓ 2 callersMethodminimize
Add operations to minimize `loss` by updating `var_list` with SGD and compute the batch size for the next step according to the CABS rule.
cabs.py:56
↓ 2 callersFunctionread_cifar10
Reads and parses examples from CIFAR10 data files. Recommendation: if you want N-way read parallelism, call this function N times. This will give
examples/cifar10_adaptive_batchsize.py:59
↓ 1 callersFunction_AddGradMom
Computes gradient moment for a bias variable through an Add operation. Assumes ``Z = tf.add(Zz, b)``, where ``b`` is a bias parameter and ``Zz``
gradient_moment.py:191
↓ 1 callersFunction_Conv2DGradMom
Computes gradient moment for the filter of a Conv2D operation. Assumes ``Z=tf.nn.conv2d(A, f)``, where ``f`` is a ``[h_f, w_f, c_in, c_out]`` c
gradient_moment.py:158
↓ 1 callersFunction_GradMom
Wrapper function for the operation type-specific GradMom functions below. Inputs: :op: A tensorflow operation of type in VALID_TYPES.
gradient_moment.py:111
↓ 1 callersFunction_MatMulGradMom
Computes gradient moment for a weight matrix through a MatMul operation. Assumes ``Z=tf.matmul(A, W)``, where ``W`` is a d1xd2 weight matrix, ``A
gradient_moment.py:134
↓ 1 callersFunction_check_and_sort_ops
Sort a list of ops according to type into valid types for which we can compute the gradient moment) and regularizers. Raise an exception when enco
gradient_moment.py:45
Method__init__
Construct a new gradient descent optimizer with coupled adaptive batch size (CABS). Args: :learning_rate: A Tensor or a floating po
cabs.py:23
Function_progress
(count, block_size, total_size)
examples/cifar10_adaptive_batchsize.py:49
Functiondistorted_inputs
Construct distorted input for CIFAR training using the Reader ops. Args: data_dir: Path to the CIFAR-10 data directory. batch_size: Number o
examples/cifar10_adaptive_batchsize.py:154
Functiongrads_and_grad_moms
Compute the gradients and gradient moments of ``loss`` w.r.t. to the variables in ``var_list`` Inputs: :loss: The tensor containing the s
gradient_moment.py:64
Functioninputs
Construct input for CIFAR evaluation using the Reader ops. Args: eval_data: bool, indicating if one should use the train or eval data set. d
examples/cifar10_adaptive_batchsize.py:211
Functionset_up_model
()
examples/models/mnist_2conv_2dense.py:23
Functionset_up_model
(images, labels)
examples/models/cifar10_2conv_3dense.py:25