Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Change name of arguments in Segmentation and Byol task and add timm support for Byol task #903

Closed
wants to merge 7 commits into from

Conversation

nilsleh
Copy link
Collaborator

@nilsleh nilsleh commented Nov 16, 2022

Change the argument name for the tasks according to comment in #854.

segmentation_model -> model
encoder_name -> encoder

Additionally, add timm model support for byol task.

@github-actions github-actions bot added testing Continuous integration testing trainers PyTorch Lightning trainers labels Nov 16, 2022
@adamjstewart adamjstewart added the backwards-incompatible Changes that are not backwards compatible label Nov 17, 2022
@adamjstewart adamjstewart added this to the 0.4.0 milestone Nov 17, 2022
@adamjstewart
Copy link
Collaborator

How hard would it be to split BYOL stuff and Segmentation stuff into separate PRs?

@nilsleh
Copy link
Collaborator Author

nilsleh commented Nov 27, 2022

How hard would it be to split BYOL stuff and Segmentation stuff into separate PRs?

Don't know but I can do that! But I think I would actually open two new separate PRs and remove this one. That feels the easiest.

@github-actions github-actions bot added the scripts Training and evaluation scripts label Nov 27, 2022
.. code-block:: python

import timm
print(timm.list_models(pretrained=True)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same here, I would suggest listing all available models:

Suggested change
print(timm.list_models(pretrained=True)
print(timm.list_models())

Also, you were missing a parenthesis

@nilsleh nilsleh closed this Nov 28, 2022
@adamjstewart adamjstewart removed this from the 0.4.0 milestone Dec 9, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backwards-incompatible Changes that are not backwards compatible scripts Training and evaluation scripts testing Continuous integration testing trainers PyTorch Lightning trainers
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants