Federated GCN for Chest X-ray screening #4
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Rakshith2597
requested changes
Feb 13, 2023
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/README.md
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/README.md
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/README.md
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/README.md
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/README.md
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/transformations.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/test/test_export.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/test/test_export.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/test/test_inference.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/test/test_train.py
Outdated
Show resolved
Hide resolved
…est_export.py Co-authored-by: Rakshith Sathish <[email protected]>
…est_inference.py Co-authored-by: Rakshith Sathish <[email protected]>
…est_train.py Co-authored-by: Rakshith Sathish <[email protected]>
Rakshith2597
changed the title
Updating the directory
Updating loss function to take weights from json file
|
|
Rakshith2597
requested changes
Feb 21, 2023
There was a problem hiding this comment.
Could you resolve the following?
- Break up train utils into 2 files. One for with GNN and another for without GNN.
- Update model weight and sample dataset links.
- Check pylint score. Details provided in the google doc task 11.
- Remove export unit tests for model with gnn.
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/inference_utils.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/model.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/model.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/train_utils.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/train_utils.py
Outdated
Show resolved
Hide resolved
misc/pytorch_toolkit/chest_xray_screening_federated_gcn/src/utils/train_utils.py
Outdated
Show resolved
Hide resolved
|
|
||
|
|
||
| #Main function for training | ||
| def trainer_with_GNN(lr, b_sz, img_pth, split_npz, train_transform, test_transform, |
…ils/inference_utils.py Co-authored-by: Rakshith Sathish <[email protected]>
…ils/model.py Co-authored-by: Rakshith Sathish <[email protected]>
…ils/model.py Co-authored-by: Rakshith Sathish <[email protected]>
…ils/train_utils.py Co-authored-by: Rakshith Sathish <[email protected]>
…ils/train_utils.py Co-authored-by: Rakshith Sathish <[email protected]>
…ils/train_utils.py Co-authored-by: Rakshith Sathish <[email protected]>
|
I removed test export for gnn. Should i do the same with test_inference for gnn. |
You can remove inference using onnx and IR for GNNs. |
|
|
Rakshith2597
approved these changes
Feb 23, 2023
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.