-
- Downloads
Added bce and mse losses
Showing
- reading_machine/include/TransitionSet.hpp 1 addition, 1 deletionreading_machine/include/TransitionSet.hpp
- reading_machine/src/TransitionSet.cpp 15 additions, 9 deletionsreading_machine/src/TransitionSet.cpp
- torch_modules/include/ConfigDataset.hpp 2 additions, 1 deletiontorch_modules/include/ConfigDataset.hpp
- torch_modules/src/ConfigDataset.cpp 6 additions, 4 deletionstorch_modules/src/ConfigDataset.cpp
- trainer/include/Trainer.hpp 16 additions, 3 deletionstrainer/include/Trainer.hpp
- trainer/src/MacaonTrain.cpp 4 additions, 1 deletiontrainer/src/MacaonTrain.cpp
- trainer/src/Trainer.cpp 72 additions, 74 deletionstrainer/src/Trainer.cpp
Loading
Please register or sign in to comment