forked from GediminasMasaitis/texel-tuner
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.h
31 lines (28 loc) · 982 Bytes
/
config.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#ifndef CONFIG_H
#define CONFIG_H 1
#include <cstdint>
#include "engines/fourku.h"
#include "engines/toy.h"
#include "engines/toy_tapered.h"
#include "engines/lynx.hpp"
#define TAPERED 1
//using TuneEval = Toy::ToyEval;
//using TuneEval = Toy::ToyEvalTapered;
//using TuneEval = Fourku::FourkuEval;
using TuneEval = Lynx;
constexpr int32_t data_load_thread_count = 8;
constexpr int32_t thread_count = 16;
constexpr tune_t preferred_k = 0;
constexpr int32_t max_epoch = 5001;
constexpr bool retune_from_zero = true;
constexpr bool enable_qsearch = false;
constexpr bool filter_in_check = false;
constexpr tune_t initial_learning_rate = 1;
constexpr int32_t learning_rate_drop_interval = 10000;
constexpr tune_t learning_rate_drop_ratio = 1;
constexpr bool print_data_entries = false;
constexpr bool print_eval = false;
constexpr bool complete_step_ouput = true;
constexpr int32_t data_load_print_interval = 10000;
constexpr int32_t print_step_interval = 1000;
#endif // !CONFIG_H