How to set batch_size, iterations and other parameters?

liuruijin17 / LSTR

This is an official repository of End-to-end Lane Shape Prediction with Transformers.

BSD 3-Clause "New" or "Revised" License

647 stars 130 forks source link

How to set batch_size, iterations and other parameters? #80

Open AvneeshKhanna opened 2 years ago

AvneeshKhanna commented 2 years ago

Hi I wanted to ask is there a way to set the number of training iterations, batch size etc. before executing the training? I wanted to tweak the training hyperparameters and run the model for preliminary results on my own dataset.

Thanks!