Starting on the model.h, but need to make layer structs and structs for loss and optimizers
This commit is contained in:
@@ -10,7 +10,7 @@
|
||||
|
||||
namespace neural_networks{
|
||||
|
||||
template <typename Td, typename Ti>
|
||||
template <typename Td, typename Ti = Td>
|
||||
struct Loss{
|
||||
|
||||
utils::Vector<Td> sample_losses;
|
||||
|
||||
Reference in New Issue
Block a user