MarekKowalski/DeepAlignmentNetwork

how to train model via the codes?

DcYond opened this issue · 3 comments

DcYond commented

@MarekKowalski Thanks for sharing the codes. how to train model via the codes?

MarekKowalski commented

You should use the DANtraining.py script. Before you do that however you need to "build" the training set by running: TrainingSetPreparation.py
You will also need to download the 300-W training set from the IBUG website beforehand.

Marek

hengshan123 commented

Hi I want to know how many epochs should each stage train? I see the num_epochs is 1000 in DANtraining.py, this just means the training is run in 1000 * 64 images?

So both of the two stages is trained by 1000 epoch? is that right?

MarekKowalski commented

Hi,

The training is ran for as long as the validation error is decreasing "fairly rapidly".
I would finish training if a lower error is not achieved after, let's say two or three hours since the last best error was achieved. The time and number of epochs this takes varies.

Marek