Skip to content
Merged
Changes from 1 commit
Commits
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
3 changes: 3 additions & 0 deletions tests/utils_framework.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@

from keras.optimizers import SGD, Adam

from keras.losses import CategoricalCrossentropy as CategoricalCrossentropy

from keras.layers import Layer as tLayer

from keras.layers import Dense as tLinear
Expand Down Expand Up @@ -146,6 +148,7 @@
"Loss",
"process_labels_for_multi_gpu",
"vanillaModel",
"CategoricalCrossentropy",
"HingeMarginLoss",
"KRMulticlassLoss",
"HingeMulticlassLoss",
Expand Down
Loading