Commit 1c6d9ec6 authored by kiegeland's avatar kiegeland
Browse files

Update README.md

parent f80c675d
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -36,8 +36,10 @@ The pretrained model is retrained onto the 80 categories from coco.

* How to retrain Singlelabel:  
Für das training mit default Hyperparameter muss nur das Verzeichnis mit den Bildern übergeben werden:  
'''python
python3 retrain.py \  
--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/  
'''
Veränderte Hyperparameter können auch als Argumente übergeben werden:  
python retrain.py \  
--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/ \