diff --git a/README.md b/README.md index 3abc605..6a267bc 100644 --- a/README.md +++ b/README.md @@ -86,11 +86,7 @@ data: # Dataset arguments from lightorch.training.cli import trainer if __name__ == '__main__': - trainer( - matmul_precision = 'medium', # default - deterministic = True, # default for reproducibility - seed = 123, # default for reproducibility - ) + trainer() ``` exec: `python3 -m training -c config.yaml` diff --git a/tests/test_supervised.py b/tests/test_supervised.py index afc3981..8d97db3 100644 --- a/tests/test_supervised.py +++ b/tests/test_supervised.py @@ -70,7 +70,7 @@ def test_supervised() -> None: datamodule=DataModule, valid_metrics="MSE", datamodule_kwargs=dict(pin_memory=False, num_workers=1, batch_size=1), - directions=["minimize"], + directions="minimize", precision="high", n_trials=10, trianer_kwargs=dict(fast_dev_run=True),