-
-
Notifications
You must be signed in to change notification settings - Fork 46.6k
Added SimCLR (Deep Learning Framework) #11900
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
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Click here to look at the relevant links ⬇️
🔗 Relevant Links
Repository:
Python:
Automated review generated by algorithms-keeper. If there's any problem regarding this review, please open an issue about it.
algorithms-keeper
commands and options
algorithms-keeper actions can be triggered by commenting on this PR:
@algorithms-keeper review
to trigger the checks for only added pull request files@algorithms-keeper review-all
to trigger the checks for all the pull request files, including the modified files. As we cannot post review comments on lines not part of the diff, this command will post all the messages in one comment.NOTE: Commands are in beta and so this feature is restricted only to a member or owner of the organization.
return (data["data"], data["target"]) | ||
|
||
|
||
def simclr_model(input_shape=(32, 32, 3), projection_dim=64) -> Model: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please provide type hint for the parameter: input_shape
Please provide type hint for the parameter: projection_dim
return Model(inputs, outputs) | ||
|
||
|
||
def contrastive_loss(projection_1, projection_2, temperature=0.1): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please provide return type hint for the function: contrastive_loss
. If the function does not return a value, please provide the type hint as: def function() -> None:
Please provide type hint for the parameter: projection_1
Please provide type hint for the parameter: projection_2
Please provide type hint for the parameter: temperature
Describe your change:
Checklist: