Commit f06cc002 authored by Rudolf Chrispens's avatar Rudolf Chrispens
Browse files

git ignore and epoch config

parent 15c1882c

.gitignore

+2 −0
Original line number Diff line number Diff line
@@ -10,6 +10,8 @@ capiton-lib/lstm/Read_Data 
capiton-lib/log



lstm/Trained_Data

lstm/Input_Data

lstm/Read_Data

log

caption-lib/lstm/lstm_main.py

+1 −1
Original line number Diff line number Diff line
@@ -56,7 +56,7 @@ def main(self, parameter_list): 
    solver = _trainer.coco_trainer( model,

                                    train_data,

                                    val_data,

                                    n_epochs=3,

                                    n_epochs=_config.trainer.num_epochs,

                                    batch_size=_config.trainer.batch_size,

                                    update_rule='adam',

                                    learning_rate=0.001,