nmt icon indicating copy to clipboard operation
nmt copied to clipboard

How can I control epoch num?

Open playma opened this issue 8 years ago • 3 comments

The README.md said that the training log will like this: # Start epoch 0, step 0, lr 1, Tue Apr 25 23:17:41 2017

but I got # Start step 0, lr 1, Thu Dec 21 10:44:55 2017

There is no epoch information, and how can I input the epoch I want to train.

playma avatar Dec 21 '17 02:12 playma

You can add a flag --num_train_steps . For example:

python -m nmt.nmt \
    ... \
    --num_train_steps=10000 \
    ... 

StupidL avatar Dec 29 '17 06:12 StupidL

@playma

In addition to the --num_train_steps, you can get some epoch info here.

oahziur avatar Jan 02 '18 20:01 oahziur

@oahziur Anyway, you may want to clarify (in the tutorial) why every epoch apparently takes 1000 steps.

nbro avatar Jan 10 '18 23:01 nbro