Argument warmup_duration should be integer, but given 0.025

[tegee]

TypeError: Argument warmup_duration should be integer, but given 0.025

[Wyldhunt]

I'm getting the exact same error
using:
python ./pretraining_train.py --dataset_path wikitext-2 --n_epochs=10

with Python 3.9.9