Added functions getDevice and setDevice
Showing
- decoder/src/MacaonDecode.cpp 3 additions, 3 deletionsdecoder/src/MacaonDecode.cpp
- reading_machine/src/Classifier.cpp 7 additions, 6 deletionsreading_machine/src/Classifier.cpp
- torch_modules/include/NeuralNetwork.hpp 3 additions, 1 deletiontorch_modules/include/NeuralNetwork.hpp
- torch_modules/src/ConfigDataset.cpp 4 additions, 4 deletionstorch_modules/src/ConfigDataset.cpp
- torch_modules/src/CustomHingeLoss.cpp 1 addition, 1 deletiontorch_modules/src/CustomHingeLoss.cpp
- torch_modules/src/LossFunction.cpp 2 additions, 2 deletionstorch_modules/src/LossFunction.cpp
- torch_modules/src/ModularNetwork.cpp 1 addition, 1 deletiontorch_modules/src/ModularNetwork.cpp
- torch_modules/src/NeuralNetwork.cpp 10 additions, 0 deletionstorch_modules/src/NeuralNetwork.cpp
- torch_modules/src/RandomNetwork.cpp 1 addition, 1 deletiontorch_modules/src/RandomNetwork.cpp
- trainer/src/MacaonTrain.cpp 5 additions, 5 deletionstrainer/src/MacaonTrain.cpp
- trainer/src/Trainer.cpp 4 additions, 4 deletionstrainer/src/Trainer.cpp
Loading
Please register or sign in to comment