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

TensorFlow v2 Support #2127

Closed
4 of 19 tasks
liuzhe-lz opened this issue Mar 6, 2020 · 2 comments
Closed
4 of 19 tasks

TensorFlow v2 Support #2127

liuzhe-lz opened this issue Mar 6, 2020 · 2 comments

Comments

@liuzhe-lz
Copy link
Contributor

liuzhe-lz commented Mar 6, 2020

Scope for July release:

  • Review on how the current NNI Pytorch model compression implementation, survey and compare the corresponding TF 2.0 supportability. ETA 7/22
  • design depends on the survey results

This issue tracks pending jobs for TensorFlow v2 support.

  • HPO
  • NAS: classic NAS tuning + ENAS, DARTS algo refresh
  • model compression
  • feature engineering

  • Examples
    • P1 - Rewrite TFv1 example trials (naive example) with TFv2 (May release)
    • P2 - Rewrite TFv1 example trials (DARTS) with TFv2
    • P0 NAS performance improvement (June)
    • P2 Move pipelines to v2
  • NAS
    • P0 - SDK and ENAS (May release)
    • Implement algorithms
  • Static graph speed-up for NAS and Model Compression
    • Ad-hoc optimization for built-in algorithms
    • Public optimizing API
  • Update building- and releasing-related scripts
  • Update documentation
@scarlett2018
Copy link
Member

released in 1.6, will keep on the rest in June release

  • P1 - Rewrite TFv1 example trials (naive example) with TFv2
  • P0 - SDK and ENAS

@scarlett2018
Copy link
Member

[3.1] performance for all, slower than pytorch, GPU usage is slow, need improvement
[3.2] test pipeline for TF2.0
[3.3] TF NAS API style refactor -> to tf style
[3.4] model compression for TF2.0
[3.5] Feature Enginerring for TF2.0
[3.6] Support all NAS algorithms for TF2.0

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

2 participants