Spaces:
Runtime error
Runtime error
File size: 2,130 Bytes
3f9d71f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
MAXIM_CONFIGS = {
# params: 6.108515000000001 M, GFLOPS: 93.163716608
"S-1": {
"features": 32,
"depth": 3,
"num_stages": 1,
"num_groups": 2,
"num_bottleneck_blocks": 2,
"block_gmlp_factor": 2,
"grid_gmlp_factor": 2,
"input_proj_factor": 2,
"channels_reduction": 4,
"name": "s1",
},
# params: 13.35383 M, GFLOPS: 206.743273472
"S-2": {
"features": 32,
"depth": 3,
"num_stages": 2,
"num_groups": 2,
"num_bottleneck_blocks": 2,
"block_gmlp_factor": 2,
"grid_gmlp_factor": 2,
"input_proj_factor": 2,
"channels_reduction": 4,
"name": "s2",
},
# params: 20.599145 M, GFLOPS: 320.32194560000005
"S-3": {
"features": 32,
"depth": 3,
"num_stages": 3,
"num_groups": 2,
"num_bottleneck_blocks": 2,
"block_gmlp_factor": 2,
"grid_gmlp_factor": 2,
"input_proj_factor": 2,
"channels_reduction": 4,
"name": "s3",
},
# params: 19.361219000000002 M, 308.495712256 GFLOPs
"M-1": {
"features": 64,
"depth": 3,
"num_stages": 1,
"num_groups": 2,
"num_bottleneck_blocks": 2,
"block_gmlp_factor": 2,
"grid_gmlp_factor": 2,
"input_proj_factor": 2,
"channels_reduction": 4,
"name": "m1",
},
# params: 40.83911 M, 675.25541888 GFLOPs
"M-2": {
"features": 64,
"depth": 3,
"num_stages": 2,
"num_groups": 2,
"num_bottleneck_blocks": 2,
"block_gmlp_factor": 2,
"grid_gmlp_factor": 2,
"input_proj_factor": 2,
"channels_reduction": 4,
"name": "m2",
},
# params: 62.317001 M, 1042.014666752 GFLOPs
"M-3": {
"features": 64,
"depth": 3,
"num_stages": 3,
"num_groups": 2,
"num_bottleneck_blocks": 2,
"block_gmlp_factor": 2,
"grid_gmlp_factor": 2,
"input_proj_factor": 2,
"channels_reduction": 4,
"name": "m3",
},
}
|