-
- Downloads
optimize SGD and ADAM update calculations. For ADAM define specific update...
optimize SGD and ADAM update calculations. For ADAM define specific update functions which are re-implemented for each architecture
Showing
- tmva/tmva/inc/TMVA/DNN/Adam.h 17 additions, 61 deletionstmva/tmva/inc/TMVA/DNN/Adam.h
- tmva/tmva/inc/TMVA/DNN/Architectures/Cpu.h 6 additions, 0 deletionstmva/tmva/inc/TMVA/DNN/Architectures/Cpu.h
- tmva/tmva/inc/TMVA/DNN/Architectures/Cuda.h 6 additions, 0 deletionstmva/tmva/inc/TMVA/DNN/Architectures/Cuda.h
- tmva/tmva/inc/TMVA/DNN/Architectures/Reference.h 9 additions, 0 deletionstmva/tmva/inc/TMVA/DNN/Architectures/Reference.h
- tmva/tmva/inc/TMVA/DNN/SGD.h 11 additions, 12 deletionstmva/tmva/inc/TMVA/DNN/SGD.h
- tmva/tmva/src/DNN/Architectures/Cpu/Arithmetic.cxx 41 additions, 0 deletionstmva/tmva/src/DNN/Architectures/Cpu/Arithmetic.cxx
- tmva/tmva/src/DNN/Architectures/Cuda/Arithmetic.cu 51 additions, 0 deletionstmva/tmva/src/DNN/Architectures/Cuda/Arithmetic.cu
- tmva/tmva/src/DNN/Architectures/Cuda/Kernels.cuh 46 additions, 1 deletiontmva/tmva/src/DNN/Architectures/Cuda/Kernels.cuh
- tmva/tmva/src/DNN/Architectures/Reference/Arithmetic.cxx 39 additions, 0 deletionstmva/tmva/src/DNN/Architectures/Reference/Arithmetic.cxx
Loading
Please register or sign in to comment