TabPFN icon indicating copy to clipboard operation
TabPFN copied to clipboard

train_mixed_precision is not passed in model_builder?

Open amueller opened this issue 1 year ago • 0 comments

It looks to me like train_mixed_precision is not passed to train in model_builder.get_model so no matter what the config_sample says, there's no mixed precision from what I can tell. Doing mixed precision training seems to give a speedup of about 30%.

amueller avatar Aug 10 '23 05:08 amueller