|
| 1 | +# Copyright (c) Alibaba, Inc. and its affiliates. |
| 2 | +# The implementation is also open-sourced by the authors, and available at |
| 3 | +# https://github.com/alibaba/lightweight-neural-architecture-search. |
| 4 | + |
| 5 | +work_dir = './save_model/E3DM_FLOPs_185e8/' |
| 6 | +log_level = 'INFO' # INFO/DEBUG/ERROR |
| 7 | +log_freq = 1000 |
| 8 | + |
| 9 | +""" video config """ |
| 10 | +image_size = 312 |
| 11 | +frames = 16 |
| 12 | + |
| 13 | +""" Model config """ |
| 14 | +model = dict( |
| 15 | + type = 'Cnn3DNet', |
| 16 | + structure_info = [ |
| 17 | + {'class': 'Conv3DKXBNRELU', 'in': 3, 'out': 24, 's': 2, 'kt': 1, 'k': 3}, \ |
| 18 | + {'class': 'SuperRes3DK1DWK1', 'in': 24, 'out': 24, 's': 2, 'kt': 1, 'k': 5, 'L': 1, 'btn': 48}, \ |
| 19 | + {'class': 'SuperRes3DK1DWK1', 'in': 24, 'out': 48, 's': 2, 'kt': 3, 'k': 3, 'L': 1, 'btn': 96}, \ |
| 20 | + {'class': 'SuperRes3DK1DWK1', 'in': 48, 'out': 96, 's': 2, 'kt': 3, 'k': 3, 'L': 1, 'btn': 192}, \ |
| 21 | + {'class': 'SuperRes3DK1DWK1', 'in': 96, 'out': 96, 's': 1, 'kt': 3, 'k': 3, 'L': 1, 'btn': 192}, \ |
| 22 | + {'class': 'SuperRes3DK1DWK1', 'in': 96, 'out': 192, 's': 2, 'kt': 3, 'k': 3, 'L': 1, 'btn': 384}, \ |
| 23 | + {'class': 'Conv3DKXBNRELU', 'in': 192, 'out': 512, 's': 1, 'kt': 1, 'k': 1},\ |
| 24 | + ] |
| 25 | +) |
| 26 | + |
| 27 | +""" Budget config """ |
| 28 | +budgets = [ |
| 29 | + dict(type = "flops", budget = 185e8), |
| 30 | + dict(type = "layers",budget = 167), |
| 31 | + ] |
| 32 | + |
| 33 | +""" Score config """ |
| 34 | +score = dict(type = 'stentr', multi_block_ratio = [0,0,0,0,1], frames=16) |
| 35 | + |
| 36 | +""" Space config """ |
| 37 | +space = dict( |
| 38 | + type = 'space_3d_k1dwk1', |
| 39 | + image_size = image_size, |
| 40 | + ) |
| 41 | + |
| 42 | +""" Search config """ |
| 43 | +search=dict( |
| 44 | + minor_mutation = False, # whether fix the stage layer |
| 45 | + minor_iter = 100000, # which iteration to enable minor_mutation |
| 46 | + popu_size = 256, |
| 47 | + num_random_nets = 100000, # the searching iterations |
| 48 | + sync_size_ratio = 1.0, # control each thread sync number: ratio * popu_size |
| 49 | + num_network = 1, |
| 50 | +) |
| 51 | + |
0 commit comments