Skip to content

Commit

Permalink
Merge pull request #83 from kozistr/fix/colab-env
Browse files Browse the repository at this point in the history
[Fix] python 3.7 for Colab envrionment
  • Loading branch information
kozistr authored Nov 1, 2022
2 parents 3c491f3 + 29114e2 commit 154a872
Show file tree
Hide file tree
Showing 6 changed files with 395 additions and 221 deletions.
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -18,4 +18,4 @@ check:

requirements:
python -m poetry export -f requirements.txt --output requirements.txt --without-hashes
python -m poetry export --dev -f requirements.txt --output requirements-dev.txt --without-hashes
python -m poetry export -f requirements.txt --output requirements-dev.txt --without-hashes --with dev
10 changes: 10 additions & 0 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,16 @@ Also, you can load the optimizer via `torch.hub`
optimizer = opt(model.parameters())


And you can check the supported optimizers & lr schedulers.

::

from pytorch_optimizer import get_supported_optimizers, get_supported_lr_schedulers

supported_optimizers = get_supported_optimizers()
supported_lr_schedulers = get_supported_lr_schedulers()


Supported Optimizers
--------------------

Expand Down
Loading

0 comments on commit 154a872

Please sign in to comment.