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

start train now

parent 86d557a6

caption-lib/lstm/Readers/helper.py

+1 −0
Original line number Diff line number Diff line
@@ -8,6 +8,7 @@ from lstm.configuration import current as config 


parent_folder_path = os.path.dirname(__file__) + "/.."

inputPath = parent_folder_path + config.path.input_folder_path

print()

print("Reader: inputPath", inputPath)

outputPath = parent_folder_path + config.path.read_folder_path

print("Reader: outputPath", outputPath)

caption-lib/lstm/Trainers/coco_trainer.py

+2 −1
Original line number Diff line number Diff line
@@ -43,7 +43,7 @@ class coco_trainer(object): 
        self.print_every = kwargs.pop('print_every', 100)

        self.save_every = kwargs.pop('save_every', 1)

        self.log_path = kwargs.pop('log_path', './log/')

        self.model_path = kwargs.pop('model_path', './model/')

        self.model_path = kwargs.pop('model_path', _config.path.model_folder_path + _config.path.model_trained)

        self.pretrained_model = kwargs.pop('pretrained_model', None)

        self.test_model = kwargs.pop('test_model', './model/lstm/model-1')
@@ -60,6 +60,7 @@ class coco_trainer(object): 
        if not os.path.exists(self.log_path):

            os.makedirs(self.log_path)





        if _config.trainer.verbose:

            print("\n#>\t INIT TRAINING")

            print("#>\ttf.Optimizer", self.optimizer)

caption-lib/lstm/configuration.py

+4 −3
Original line number Diff line number Diff line
@@ -18,9 +18,9 @@ class default_paths(object): 
    read_test = "/coco.test"

    # model state saves

    model_folder_path = "/Trained_Data"

    model_trained = "/3/trained_model"

    model_trained = "/1"

    # model for validation

    model_for_validation = "/1/trained_model"

    model_for_validation = "/1/model-10"





class default_reader(object):  # default reader config
@@ -47,10 +47,11 @@ class default_trainer(object): 
    num_steps = 35

    vector_size = 2048

    batch_size = 20

    dropout_on = True

    dropout = 0.5

    init_scale = 0.05

    forget_bias = 1.0

    hidden_size = 2048

    hidden_size = 1024

    learning_rate = 0.2

caption-lib/lstm/lstm_main.py

+6 −5
Original line number Diff line number Diff line
@@ -24,7 +24,8 @@ def main(self, parameter_list): 
        trained_model = reader.helper.modelPath + config.path.model_for_validation + "-final"

        current_trainer.test(trained_model, test_data, reversed_dictionary)

    """



    #check folder paths

    _reader.helper.check_folders()



    # load train dataset

    current_reader = _reader.coco_reader()
@@ -42,13 +43,13 @@ def main(self, parameter_list): 
                                current_reader,

                                dim_feature=[_config.trainer.batch_size, _config.trainer.vector_size],

                                dim_embed=_config.trainer.vector_size,

                                dim_hidden=1024,

                                dim_hidden=_config.trainer.hidden_size,

                                n_time_step=16,

                                prev2out=True,

                                ctx2out=True,

                                alpha_c=1.0,

                                selector=True,

                                dropout=True)

                                dropout=_config.trainer.dropout_on)





    solver = _trainer.coco_trainer( model,
@@ -62,8 +63,8 @@ def main(self, parameter_list): 
                                    save_every=2,

                                    image_path='./image/',

                                    pretrained_model=None,

                                    model_path='Trained_Data/lstm/',

                                    test_model='Trained_Data/lstm/model-10',

                                    model_path=_reader.helper.modelPath + _config.path.model_trained,

                                    test_model=_reader.helper.modelPath + _config.path.model_for_validation,

                                    print_bleu=False,

                                    log_path='log/')

caption-lib/start.py

+1 −1
Original line number Diff line number Diff line 
import lstm



#clear console

print("\033c")

#print("\033c")



#start proj

lstm.lstm_main.main("Start", None)
 
 No newline at end of file