-
Notifications
You must be signed in to change notification settings - Fork 136
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
model: pytorch: Add PyTorch based pre-trained ConvNet models #784
Conversation
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
This pull request introduces 3 alerts when merging ffe195a into b7e9d30 - view on LGTM.com new alerts:
|
Codecov Report
@@ Coverage Diff @@
## master #784 +/- ##
=======================================
Coverage 87.45% 87.45%
=======================================
Files 130 130
Lines 7897 7897
Branches 1269 1269
=======================================
Hits 6906 6906
Misses 752 752
Partials 239 239 Continue to review full report at Codecov.
|
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
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.
Since this is similar to model/scikit
let's copy what was done in the __init__.py
into model/pytorch/dffml_model_pytorch/__init__.py
and modify accordingly
else self._model.fc.parameters() | ||
) | ||
|
||
self.criterion = nn.CrossEntropyLoss() |
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.
We'll follow #806 guide once it's complete to make these customizable
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
This pull request introduces 1 alert when merging a2a2872 into 6a26289 - view on LGTM.com new alerts:
|
Signed-off-by: sakshamarora1 <sakshamarora1001@gmail.com>
This pull request introduces 1 alert when merging e8acc4f into 6a26289 - view on LGTM.com new alerts:
|
Closes: #765
Signed-off-by: sakshamarora1 sakshamarora1001@gmail.com