Graphormer icon indicating copy to clipboard operation
Graphormer copied to clipboard

Customizing number of epochs for training.

Open aahad5555 opened this issue 3 years ago • 3 comments

Hi, I have a quick question that how can I customize the number of epochs for training?

aahad5555 avatar Dec 27 '21 07:12 aahad5555

You can use --max-epoch to specify the number of epoches. Since Graphormer is based on Fairseq now, more training options can be found in FairSeq's DOC.

GYHHAHA avatar Dec 27 '21 08:12 GYHHAHA

Thank you @GYHHAHA for your timely reply.

aahad5555 avatar Dec 27 '21 09:12 aahad5555

Both --total-num-update and --max-epoch could be used for specifying training steps.

zhengsx avatar Dec 27 '21 12:12 zhengsx