Number of training epochs - Githubissues

tilemmpon / Singapore-Maritime-Dataset-Trained-Deep-Learning-Models

This repository contains the training configurations for several Deep Learning models trained on the Singapore Maritime Dataset and links to the trained - ready to use - models.

36 stars 14 forks source link

Number of training epochs #8

Open darrenjkt opened 5 months ago

darrenjkt commented 5 months ago

Hi thanks for your models, I was wondering how many epochs did you train your models on for this dataset?

FengJunJian commented 5 months ago

您好，已收到您的邮件。

tilemmpon commented 2 months ago

I apologize for the late response. The epochs are included in the respective configuration files.

For YOLO see an example here: https://github.com/tilemmpon/Singapore-Maritime-Dataset-Trained-Deep-Learning-Models/blob/6dd31b2f2f31358be4ea1e32b1d247fae4f84184/YOLOv2_training_configurations/config_full_yolo.json#L18
For TensorFlow epochs depend on the steps see an example here: https://github.com/tilemmpon/Singapore-Maritime-Dataset-Trained-Deep-Learning-Models/blob/6dd31b2f2f31358be4ea1e32b1d247fae4f84184/tensorflow_training_configurations/faster_rcnn_inception_v2_smd.config#L93 but I cannot recall how they correlate exactly.