Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Add self-supervised learning capabilities to InnerEye #440

Merged
merged 403 commits into from
May 7, 2021
Merged

Conversation

melanibe
Copy link
Contributor

@melanibe melanibe commented Apr 20, 2021

This PR adds support for training of self-supervised models in InnerEye. The implementation is based on the new bring-your-own model framework. We provide two examples to get started with training of SSL models: CIFAR10 SSL training and Chest-Xrays based models. See SSL doc for more details.

ant0nsc and others added 30 commits April 1, 2021 15:14
…aModule for all datasets, make CIFAR follow the same path as CXR, add NIH support
…DeepLearning into melanibe/ssl

� Conflicts:
�	InnerEye/ML/run_ml.py
�	InnerEye/ML/runner.py
�	Tests/ML/configs/lightning_test_containers.py
�	Tests/ML/test_lightning_containers.py
docs/self_supervised_models.md Outdated Show resolved Hide resolved
docs/self_supervised_models.md Outdated Show resolved Hide resolved
ozan-oktay
ozan-oktay previously approved these changes May 7, 2021
Copy link
Contributor

@ozan-oktay ozan-oktay left a comment

Choose a reason for hiding this comment

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

👍

Shruthi42
Shruthi42 previously approved these changes May 7, 2021
@melanibe melanibe dismissed ant0nsc’s stale review May 7, 2021 12:29

Addressed changes

@melanibe melanibe enabled auto-merge (squash) May 7, 2021 13:03
@melanibe melanibe merged commit 7b5b414 into main May 7, 2021
@melanibe melanibe deleted the melanibe/ssl branch May 7, 2021 13:27
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants