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

Fix a few bugs in Retiarii and upgrade Dockerfile #3713

Merged
merged 6 commits into from
Jun 3, 2021

Conversation

ultmaster
Copy link
Contributor

@ultmaster ultmaster commented Jun 2, 2021

  1. Upgrade dependencies in Dockerfile and include pytorch-lightning.
  2. Support export in pure-python execution engine, update docs and examples.
  3. Fix an issue that nni.retiarii.nn cannot be imported in generated code.
  4. Change the RL implementation to multithreading because multiprocessing doesn't work with current execution engine implementation. As a result, we don't support AsyncCollector for now.

This PR should unblock #3700.

@ultmaster ultmaster changed the title Fix a few bugs in Retiarii Fix a few bugs in Retiarii and upgrade Dockerfile Jun 2, 2021
@@ -317,7 +318,7 @@ def export_top_models(self, top_k: int = 1, optimize_mode: str = 'maximize', for
"""
Export several top performing models.

For one-shot algorithms, only top-1 is supported. For others, ``optimize_mode`` asnd ``formater`` is
For one-shot algorithms, only top-1 is supported. For others, ``optimize_mode`` asnd ``formatter`` is
Copy link
Contributor

Choose a reason for hiding this comment

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

asnd -> and, is -> are

@ultmaster ultmaster closed this Jun 3, 2021
@ultmaster ultmaster reopened this Jun 3, 2021
@ultmaster ultmaster merged commit a284f71 into microsoft:master Jun 3, 2021
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.

3 participants