File tree Expand file tree Collapse file tree 1 file changed +5
-6
lines changed
src/pytorch_tabular/config Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -358,8 +358,8 @@ class TrainerConfig:
358358
359359 progress_bar (str): Progress bar type. Can be one of: `none`, `simple`, `rich`. Defaults to `rich`.
360360
361- precision (int ): Precision of the model. Can be one of: `32`, `16`, `64`. Defaults to `32`..
362- Choices are: [`32`,`16`,`64`].
361+ precision (str ): Precision of the model. Defaults to `32`. See
362+ https://lightning.ai/docs/pytorch/stable/common/trainer.html#precision
363363
364364 seed (int): Seed for random number generators. Defaults to 42
365365
@@ -543,11 +543,10 @@ class TrainerConfig:
543543 default = "rich" ,
544544 metadata = {"help" : "Progress bar type. Can be one of: `none`, `simple`, `rich`. Defaults to `rich`." },
545545 )
546- precision : int = field (
547- default = 32 ,
546+ precision : str = field (
547+ default = "32" ,
548548 metadata = {
549- "help" : "Precision of the model. Can be one of: `32`, `16`, `64`. Defaults to `32`." ,
550- "choices" : [32 , 16 , 64 ],
549+ "help" : "Precision of the model. Defaults to `32`." ,
551550 },
552551 )
553552 seed : int = field (
You can’t perform that action at this time.
0 commit comments