PiotrNawrot / nanoT5

Fast & Simple repository for pre-training and fine-tuning T5-style models
Apache License 2.0
970 stars 74 forks source link

How to run on CPU #18

Closed ratan-prasad closed 1 year ago

ratan-prasad commented 1 year ago

Do you any code example how to run it on CPUs

PiotrNawrot commented 1 year ago

Excuse me the delay, you should do device=cpu. Also look out for other hyperparameters to set in the config file