search

Multi-Objective-NAS / self-supervised-nas

Official implementation of the paper "Pretraining Neural Architecture Search Controllers with Locality-based Self-Supervised Learning" (NeurIPSW 2020)
5 stars 0 forks source link

Feature/#10 graph modifier testcode #13

Closed Naruu closed 3 years ago

Naruu commented 3 years ago

  1. datasets.py

    • 101, 201을 모두 사용하기 위해, ModelSpec parameter 추가 \ ( NASBench class에 (engine, ModelSpec) 또는 (api, path)를 전달해야하는데 path보다는 전자가 나을 것 같아서)
    • is_valid 함수 수정

  2. graph_modifier.py

    • graph modify가 불가능한 경우를 위해 \ MAX_GRAPH_MODIFY_TRY번 graph modify를 시도하고 실패하는 경우 modify하지 않은 원래 matrix, ops return하도록 수정
    ex) graph modify가 불가능한 경우
[[0,1],
 [1, 0]]

  3. 아직 테스트 코드 및 로직 수정은 적용되지 않았습니다. 구조 변경 먼저 PR 합니다~