#pragma once #include #define DATA_DIR "../data" #define OUT_DIR "./out" #define MODEL_DIR "./models" #ifdef __CUDACC__ #define NB_THREADS 1024 #define NB_THREADS_2D_X 32 #define NB_THREADS_2D_Y 32 #define NB_THREADS_3D_X 16 #define NB_THREADS_3D_Y 16 #define NB_THREADS_3D_Z 4 #define M static_cast(log2f(NB_THREADS_2D_Y)) #endif // Save state to avoid recalculation on restart #define SAVE_STATE true // Redo the state even if it's already saved #define FORCE_REDO false // Use GPU to greatly accelerate runtime #define GPU_BOOSTED true // Depending on what you set, the output label will be as follow : // ┌─────────────┬───────┐ // │ GPU_BOOSTED │ LABEL │ // ├─────────────┼───────┤ // │ true │ GPU │ // │ false │ CPU │ // └─────────────┴───────┘ // Number of weak classifiers // [[maybe_unused]] constexpr const std::array TS{ 1 }; // [[maybe_unused]] constexpr const std::array TS{ 1, 5, 10 }; [[maybe_unused]] constexpr const std::array TS{ 1, 5, 10, 25, 50 }; // [[maybe_unused]] constexpr const std::array TS{ 1, 5, 10, 25, 50, 100, 200, 300 }; // [[maybe_unused]] constexpr const std::array TS{ 1, 5, 10, 25, 50, 100, 200, 300, 400, 500, 1000 }; // Enable verbose output (for debugging purposes) #define __DEBUG false // Debugging options #if __DEBUG #define IDX_INSPECT 4548 #define IDX_INSPECT_OFFSET 100 #endif