- package for neural network algorithm training and tuning
uproot
+lmdb
+keras/keras-tuner
- [email protected]
-
install anaconda
-
then restore the environments
conda env create -f CondaEnv.yml
- (updated on 04/06/2020 using)
conda env export > CondaEnv.yml
- git clone this repository
# to train with static architecture
python NNAlgs Train NNAlgs/config/ConfDSNN.yaml
# to tune hyper-parameters
python NNAlgs Tune NNAlgs/config/ConfDSNN.yaml
.
├── CondaEnv.yml
├── README.md
├── __main__.py
├── config
│ ├── ArchDSNN.yaml
│ ├── ArchLSTM.yaml
│ ├── ConfDSNN.yaml
│ ├── ConfLSTM.yaml
│ ├── DatasetBuilder.py
│ ├── GeneratorLoaders.py
│ └── __init__.py
├── data
│ ├── json
│ └── lmdb
├── nnalgs
│ ├── __init__.py
│ ├── algs
│ │ ├── DataGenerators.py
│ │ ├── LMDBCreators.py
│ │ ├── Models.py
│ │ └── __init__.py
│ ├── base
│ │ ├── BuilderDirector.py
│ │ ├── DataGenerator.py
│ │ ├── DatasetBuilder.py
│ │ ├── GeneratorLoader.py
│ │ ├── IDataset.py
│ │ ├── LMDBCreator.py
│ │ └── __init__.py
│ └── utils
│ ├── CallBack.py
│ ├── Common.py
│ ├── Enum.py
│ ├── LMDBGen.py
│ ├── Logger.py
│ ├── Math.py
│ └── __init__.py
└── nnexec
├── TrainDecayMode.py
├── TuneDecayMode.py
└── __init__.py
- increase GPU usage [1]