Commit a5ad084d authored by chrispens's avatar chrispens
Browse files

Update README.md

parent 444c57c6

README.md

+19 −0
Original line number Diff line number Diff line
@@ -171,6 +171,25 @@ Um das komplette LSTM verwenden zu können sind die mscocotools notwendig. Des w 
* ipywidgets



Um das LSTM zu konfigurieren editiert man die "current settings" in der [configuration.py](/caption_lib/lstm/configuration.py)

```

Default trainingsconfig

class default_trainer(object):

    train_on = False

    train_fresh_and_override = True

    verbose = False

    num_layers = 1

    num_epochs = 60

    num_steps = 35

    vector_size = 2048

    batch_size = 20

    dropout_on = True

    dropout = 0.5

    forget_bias = 0.4

    hidden_size = 1024

    learning_rate = 0.001

    batch_randomization = True

```





Zum starten des LSTMS einfach in den ordner caption_lib navigieren und von dort aus [start.py](/caption_lib/start.py) ausführen.