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

Add NAS-bench-201 #2

Closed juice500ml closed 3 years ago

young917 commented 3 years ago

image

CUDA_VISIBLE_DEVICES=0 python3 pretrain.py experiment=TripletMarginLoss controller=nasbench201 dataset=nasbench201 코드를 돌려보았습니다...!

young917 commented 3 years ago

돌려본 결과입니다 그런데 길이가 7인 architecture가 너무 적었나는 생각이 듭니다..!

young917 commented 3 years ago

get_dataset 에서 name 과 path 를 가지고 api engine 을 만든 뒤에 NASBench object 를 만들어서 반환하는 식으로 nasbench201 branch 를 따랐습니다.

__iter__ 부분에서 master를 따라 graph_modifier 에게서 변환된 graph 를 받아 dataset 을 구성하도록 하였습니다.

_encode 부분에서nasbench201을 따라 semiNAS의 nasbench201 branch에서 짠 코드대로 convert_arch_to_seq에서 argument 가 추가되었습니다.