-
Notifications
You must be signed in to change notification settings - Fork 176
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
how to test the SP-NAS mAP on COCO with vega #14
Comments
Hi, if you want to reproduce the best performance of SP-NAS on COCO, you should get ImageNet pretrained weight of the corresponding model code in the paper (1111-21111-2111111111111111111111111-211-1), and use the following config file to replace
|
@chanyn @zhangjiajin I have two questions
general:
worker:
timeout: 1000
gpus_per_job: -1
#pipeline: [nas1, nas2]
pipeline: [nas1, nas2, fullytrain]
nas1:
pipe_step:
type: SpNasPipeStep
search_algorithm:
type: SpNas
codec: SpNasCodec
total_list: 'total_list_s.csv'
sample_level: 'serial'
max_sample: 10
max_optimal: 5
serial_settings:
num_mutate: 3
addstage_ratio: 0.05
expend_ratio: 0.3
max_stages: 6
regnition: False
# last_search_result:
search_space:
type: SearchSpace
config_template_file: "./nas/sp_nas/cascade_rcnn_r50_fpn_1x.py"
epoch: 5
trainer:
type: SpNasTrainer
gpus: 8
debug: False
nas2:
pipe_step:
type: SpNasPipeStep
search_algorithm:
type: SpNas
codec: SpNasCodec
total_list: 'total_list_p.csv'
sample_level: 'parallel'
max_sample: 10
max_optimal: 1
serial_settings:
last_search_result: 'total_list_s.csv'
regnition: False
search_space:
type: SearchSpace
config_template_file: "./nas/sp_nas/cascade_rcnn_r50_fpn_1x.py"
epoch: 5
trainer:
type: SpNasTrainer
gpus: 8
debug: False
fullytrain:
pipe_step:
type: FullyTrainPipeStep
trainer:
type: SpNasTrainer
gpus: 8
model_desc_file: 'total_list_p.csv'
# # model_desc: 'r_111-2111-211111-211_0-0-0-0'
config_template: "./nas/sp_nas/cascade_rcnn_r50_fpn_1x.py"
regnition: False
epoch: 12
debug: False
|
how to test the SP-NAS mAP on COCO with vega
The text was updated successfully, but these errors were encountered: