[ADD] Application of RNN/LSTM `param_jac_t` for a subset of samples #197

f-dangel · 2021-07-05T11:34:56Z

Prepares the core functionality to support #12.

backpack/core/derivatives/basederivatives.py

f-dangel and others added 8 commits July 2, 2021 18:47

[ADD] Extend weight_jac_t by subsampling argument

886e3d0

[ADD] Sub-sampling for batch normalization, adapt interface

83d1070

[REF] Separate fixtures for filtering weights and matrix generation

77652b6

[REF] Shorten instantiation fixture name

ec6126c

[TEST] Add BN setting to weight_jac_t tests

539aa54

[DOC] Add batch norm test setting to fully-documented

92bbdf7

[ADD] Support subsampling in bias_jac_t

f2f0827

[ADD] Support sub-sampling in RNN/LSTM param_weight_jac_t

ce9dba5

f-dangel mentioned this pull request Jul 5, 2021

Mini-batch subsampling #12

Closed

f-dangel added 2 commits July 6, 2021 19:07

Merge branch 'development' into subsampling-rnn-lstm-param-jac-t

6ca9bc6

[FIX] flake8

dafb46f

f-dangel commented Jul 6, 2021

View reviewed changes

backpack/core/derivatives/basederivatives.py Outdated Show resolved Hide resolved

f-dangel commented Jul 6, 2021

View reviewed changes

backpack/core/derivatives/basederivatives.py Outdated Show resolved Hide resolved

f-dangel commented Jul 6, 2021

View reviewed changes

backpack/core/derivatives/basederivatives.py Outdated Show resolved Hide resolved

f-dangel commented Jul 6, 2021

View reviewed changes

backpack/core/derivatives/basederivatives.py Outdated Show resolved Hide resolved

f-dangel added 3 commits July 6, 2021 19:17

[FMT] Squeeze some lines

2fdd703

[FIX] Typo in exception message

b6e8eb4

[DOC] Correct shapes

0704aae

f-dangel merged commit cbee344 into development Jul 6, 2021

f-dangel deleted the subsampling-rnn-lstm-param-jac-t branch July 6, 2021 17:57

Provide feedback