Bug in train_mamba.py line 53
vmajor opened this issue · 2 comments
vmajor commented
Default code generates:
~/site-packages/transformers/trainer.py", line 1597, in _inner_training_loop
max_steps = math.ceil(args.num_train_epochs * num_update_steps_per_epoch)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: must be real number, not str
Changing the line 53 to int corrects this error: parser.add_argument("--num_epochs", type=int, default=1
rwl4 commented
This should be fixed now.
justusmattern27 commented