Skip to content
Merged
Changes from all commits
Commits
Show all changes
51 commits
Select commit Hold shift + click to select a range
70b927d
default config
NihalHarish Aug 18, 2020
63ee466
add test
NihalHarish Aug 18, 2020
171522a
add mxnet and pytorch tests
NihalHarish Aug 18, 2020
1865f27
differentiate between tf and other frameworks
NihalHarish Aug 19, 2020
82f2bd6
should save layer
NihalHarish Aug 19, 2020
5762c53
handle kernelGrad
NihalHarish Aug 19, 2020
e83e126
init
NihalHarish Aug 19, 2020
63a9cad
run only for grad tape
NihalHarish Aug 19, 2020
230177b
str cast
NihalHarish Aug 19, 2020
bef5e61
scan
NihalHarish Aug 19, 2020
5204cf2
fix
NihalHarish Aug 20, 2020
b878e5f
update
NihalHarish Aug 20, 2020
744e627
limit gradient loop
NihalHarish Aug 20, 2020
c6765ca
modify assert
NihalHarish Aug 20, 2020
ade21de
Merge remote-tracking branch 'origin' into default_hook_config_gamma
NihalHarish Aug 20, 2020
dc0b0f1
logs is None
NihalHarish Aug 20, 2020
1191fa5
PR comments
NihalHarish Aug 21, 2020
85e69e5
save tensor in custom collection
NihalHarish Aug 21, 2020
6808cb7
register hook only for gradtape
NihalHarish Aug 23, 2020
bed51bc
attach layer hook only for grad tape
NihalHarish Aug 23, 2020
800b65c
register model
NihalHarish Aug 23, 2020
683c758
make tf log compliant
NihalHarish Aug 23, 2020
aeb2ba1
update tf2_2
NihalHarish Aug 23, 2020
311b131
unpack input
NihalHarish Aug 24, 2020
777567d
nit: correct comments
NihalHarish Aug 24, 2020
fdd537c
retrigger CI
NihalHarish Aug 24, 2020
469c5dd
retrigger CI
NihalHarish Aug 24, 2020
9f25d63
revert assert change
NihalHarish Aug 24, 2020
6c4ec8f
save gradients
NihalHarish Aug 24, 2020
03e6d26
address ordering bug
NihalHarish Aug 26, 2020
0613b32
retrigger CI
NihalHarish Aug 26, 2020
bdb35bd
refactor
NihalHarish Aug 26, 2020
6f570b4
merge origin
NihalHarish Aug 26, 2020
b3a3ccd
Revert "Add ability to only save shapes of tensors (#328)"
NihalHarish Aug 26, 2020
5ff1266
rename fns
NihalHarish Aug 27, 2020
0bbe25b
gradients refactor
NihalHarish Aug 28, 2020
7c5312c
revert pr 328-shapes
NihalHarish Aug 28, 2020
c0682da
test should_save_tensor
NihalHarish Aug 28, 2020
02a026c
reduce code dupe
NihalHarish Aug 28, 2020
368feb4
retrigger CI
NihalHarish Aug 28, 2020
a49a406
loop over tuple
NihalHarish Aug 28, 2020
7d8bb23
merge master
NihalHarish Aug 28, 2020
4de3e63
extract v name
NihalHarish Aug 28, 2020
4f7efae
extract v name
NihalHarish Aug 28, 2020
6861bc0
fix bug in zcc test
NihalHarish Aug 29, 2020
9d9359b
comments
NihalHarish Aug 30, 2020
b4b6924
rename fn
NihalHarish Aug 30, 2020
c42d207
retrigger CI
NihalHarish Aug 31, 2020
03e2922
retrigger CI
NihalHarish Aug 31, 2020
0a74b65
convert str to bytes
NihalHarish Aug 31, 2020
e146f53
Merge branch 'master' into default_hook_config
NihalHarish Aug 31, 2020
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions smdebug/tensorflow/keras.py
Original file line number Diff line number Diff line change
Expand Up @@ -452,6 +452,8 @@ def save_gradients_from_logs(self, gradients):
layer_name = str(v.numpy(), "utf-8")
elif isinstance(v, tf.Variable):
layer_name = v.name
elif isinstance(v, bytes):
layer_name = str(v, "utf-8")
else:
layer_name = v
layer_name = layer_name.split(":")[0]
Expand Down