Functions — github.com/KGrewal1/candle-optimisers

hub / github.com/KGrewal1/candle-optimisers / functions

Functions178 in github.com/KGrewal1/candle-optimisers

⨍ Functions 178◇ Types & classes 63

↓ 81 callersMethodforward

(&self)

tests/lbfgs_tests.rs:38

↓ 63 callersMethodbackward_step

(&mut self, loss: &Tensor)

src/lbfgs.rs:152

↓ 12 callersMethodvars

(&self)

tests/lbfgs_tests.rs:34

↓ 9 callersFunctionadd_grad

(vs: &mut Vec<Var>, flat_tensor: &Tensor)

src/lbfgs.rs:451

↓ 6 callersMethodloss

(&self)

tests/lbfgs_tests.rs:18

↓ 4 callersMethodinner_step

( &self, params: &ParamsRMSprop, grads: &candle_core::backprop::GradStore, )

src/rmsprop.rs:100

↓ 4 callersMethodinto_inner

(self)

src/rmsprop.rs:96

↓ 4 callersMethodloss

(&self)

src/lbfgs.rs:487

↓ 3 callersMethoddirectional_evaluate

( &mut self, mag: f64, direction: &Tensor, )

src/lbfgs/strong_wolfe.rs:352

↓ 3 callersMethodinto_inner

(self)

src/esgd.rs:360

↓ 3 callersMethodl2_reg

(&self)

src/lbfgs/strong_wolfe.rs:393

↓ 2 callersFunctioncubic_interpolate

ported from pytorch torch/optim/lbfgs.py ported from <https://github.com/torch/optim/blob/master/polyinterp.lua>

src/lbfgs/strong_wolfe.rs:8

↓ 2 callersFunctionflat_grads

(vs: &Vec<Var>, loss: &Tensor, weight_decay: Option<f64>)

src/lbfgs.rs:425

↓ 2 callersMethodforward

(&self, xs: &Tensor)

examples/mnist/models.rs:28

↓ 2 callersMethodinner_step

( &self, params: &ParamsAdam, grads: &candle_core::backprop::GradStore, t: f64

src/adam.rs:118

↓ 2 callersMethodinto_inner

(self)

src/adam.rs:114

↓ 2 callersFunctionload_data

()

benches/mnist_bench.rs:14

↓ 2 callersMethodloss

(&self)

benches/training.rs:137

↓ 2 callersMethodset_params

Set the parameters for the optimiser # Warning As the AMSGrad variant requires having tracked an additional tensor this variable cannot be changed o

src/adam.rs:404

↓ 1 callersMethodback_step

(&mut self, loss: &Tensor)

benches/training.rs:27

↓ 1 callersMethodback_step

(&mut self, loss: &Tensor)

examples/mnist/optim.rs:30

↓ 1 callersMethodforward

(&self)

benches/training.rs:129

↓ 1 callersMethodforward

(&self, xs: &Tensor)

src/lbfgs.rs:512

↓ 1 callersMethodforward

(&self, xs: &Tensor)

src/lbfgs/strong_wolfe.rs:456

↓ 1 callersMethodinto_inner

(self)

src/radam.rs:251

↓ 1 callersMethodinto_inner

(self)

src/lbfgs.rs:418

↓ 1 callersMethodinto_inner

(self)

src/nadam.rs:240

↓ 1 callersMethodinto_inner

(self)

src/adagrad.rs:188

↓ 1 callersMethodinto_inner

(self)

src/adadelta.rs:199

↓ 1 callersMethodinto_inner

(self)

src/adamax.rs:199

↓ 1 callersFunctionlinear_z

(in_dim: usize, out_dim: usize, vs: &VarBuilder)

examples/mnist/models.rs:7

↓ 1 callersMethodloss

(&self)

src/lbfgs/strong_wolfe.rs:431

↓ 1 callersMethodset_betas

set the betas this can be combined with set_lr for LR and momentum decay scheduling

src/adam.rs:430

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/radam.rs:233

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/lbfgs.rs:414

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/esgd.rs:342

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/nadam.rs:222

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/adagrad.rs:170

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/adadelta.rs:181

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/rmsprop.rs:508

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/adamax.rs:181

↓ 1 callersMethodset_learning_rate

(&mut self, lr: f64)

src/adam.rs:388

↓ 1 callersMethodset_params

(&mut self, config: Self::Config)

src/radam.rs:243

↓ 1 callersMethodset_params

(&mut self, config: Self::Config)

src/esgd.rs:352

↓ 1 callersMethodset_params

(&mut self, config: Self::Config)

src/nadam.rs:232

↓ 1 callersMethodset_params

(&mut self, config: Self::Config)

src/adagrad.rs:180

↓ 1 callersMethodset_params

(&mut self, config: Self::Config)

src/adadelta.rs:191

↓ 1 callersMethodset_params

(&mut self, config: Self::Config)

src/adamax.rs:191

↓ 1 callersFunctionset_vs

(vs: &mut [Var], vals: &Vec<Tensor>)

src/lbfgs.rs:464

↓ 1 callersMethodstrong_wolfe

( &mut self, mut step_size: f64, // step size direction: &Tensor, // direction

src/lbfgs/strong_wolfe.rs:72

Functionadadelta_decoupled_weight_decay_test

()

tests/adadelta_tests.rs:104

Functionadadelta_test

()

tests/adadelta_tests.rs:40

Functionadadelta_weight_decay_test

()

tests/adadelta_tests.rs:70

Functionadagrad_decoupled_weight_decay_test

()

tests/adagrad_tests.rs:190

Functionadagrad_lr_decay_test

()

tests/adagrad_tests.rs:96

Functionadagrad_test

()

tests/adagrad_tests.rs:37

Functionadagrad_weight_decay_test

()

tests/adagrad_tests.rs:155

Functionadam_amsgrad_decay_test

()

tests/adam_tests.rs:262

Functionadam_amsgrad_test

()

tests/adam_tests.rs:206

Functionadam_test

()

tests/adam_tests.rs:40

Functionadam_weight_decay_test

()

tests/adam_tests.rs:93

Functionadamax_decoupled_weight_decay_test

()

tests/adamax_tests.rs:126

Functionadamax_test

()

tests/adamax_tests.rs:37

Functionadamax_weight_decay_test

()

tests/adamax_tests.rs:93

Functionadamw_amsgrad_decay_test

()

tests/adam_tests.rs:319

Functionadamw_weight_decay_test

()

tests/adam_tests.rs:149

Functioncriterion_benchmark_lbfgs

(c: &mut Criterion)

benches/mnist_bench.rs:68

Functioncriterion_benchmark_std

(c: &mut Criterion)

benches/mnist_bench.rs:19

create a new optimiser from a slice of variables, setup parameters and a model

src/lib.rs:132

Functioninto_inner_test

()

src/radam.rs:287

Functioninto_inner_test

()

src/lbfgs.rs:533

Functioninto_inner_test

()

src/esgd.rs:396

Functioninto_inner_test

()

src/nadam.rs:276

Functioninto_inner_test

()

src/adagrad.rs:224

Functioninto_inner_test

()

src/adadelta.rs:235

Functioninto_inner_test

()

src/rmsprop.rs:559

Functioninto_inner_test

()

src/adamax.rs:235

Functioninto_inner_test

src/lbfgs/strong_wolfe.rs:463

Functionlbfgs_rms_grad_test

()

tests/lbfgs_tests.rs:121

Functionlbfgs_rms_step_test

()

tests/lbfgs_tests.rs:159

Functionlbfgs_test

()

tests/lbfgs_tests.rs:46

Functionlbfgs_test_strong_wolfe

()

tests/lbfgs_tests.rs:83

Functionlbfgs_test_strong_wolfe_weight_decay

()

tests/lbfgs_tests.rs:198

Functionlbfgs_test_weight_decay

()

tests/lbfgs_tests.rs:238

next →1–100 of 178, ranked by callers