Save
Maybe backward on activation+loss works
This commit is contained in:
@@ -67,7 +67,7 @@ struct loss_categorical_crossentropy: loss{
|
||||
*/
|
||||
bool forward(
|
||||
const panic::tensor::real_matrix& y_pred,
|
||||
const panic::tensor::uint_vector& y_true);
|
||||
const panic::tensor::uint_vector& y_true)override;
|
||||
|
||||
/**
|
||||
* @brief forward function to calculate losses
|
||||
@@ -80,7 +80,7 @@ struct loss_categorical_crossentropy: loss{
|
||||
*/
|
||||
bool forward(
|
||||
const panic::tensor::real_matrix& y_pred,
|
||||
const panic::tensor::real_matrix& y_true);
|
||||
const panic::tensor::real_matrix& y_true)override;
|
||||
|
||||
|
||||
/**
|
||||
|
||||
Reference in New Issue
Block a user