mirror of
https://github.com/augustin64/projet-tipe
synced 2025-02-02 19:39:39 +01:00
Update train.h parameters
This commit is contained in:
parent
bb0c709a50
commit
00ac29b1d0
@ -5,9 +5,9 @@
|
|||||||
#define DEF_TRAIN_H
|
#define DEF_TRAIN_H
|
||||||
|
|
||||||
#define EPOCHS 10
|
#define EPOCHS 10
|
||||||
#define BATCHES 500
|
#define BATCHES 32
|
||||||
#define USE_MULTITHREADING
|
#define USE_MULTITHREADING
|
||||||
#define LEARNING_RATE 0.05
|
#define LEARNING_RATE 3e-4
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
Loading…
Reference in New Issue
Block a user