zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.2k stars 1.27k forks source link

KeyError: 'category_id' #579

Open valterfsj opened 3 years ago

valterfsj commented 3 years ago

Hello there,

I've been struggling to start training due my dataset/annotations. I currently have my imgs (2k+) annotated in YOLO txt format and tried to convert it to COCO JSON unsucessfully, then, tried to re-label it on COCO with a small dataset to "try" the efficientdet. My Dataset is private and I can't share IT, but, I can share the resulting JSON, as it seems that is the root cause for KeyError: 'category_id' error.

The error is the following:

(effdetpy) E:\PorosityDetection\Yet-Another-EfficientDet-Pytorch-master\Yet-Another-EfficientDet-Pytorch-master>python train.py -c 2 -p PorosityDetection --batch_size 8 --lr 1e-3 --num_epochs 10 --load_weights E:/PorosityDetection/Yet-Another-EfficientDet-Pytorch-master/Yet-Another-EfficientDet-Pytorch-master/weights/efficientdet-d8.pth --head_only True loading annotations into memory... Done (t=0.00s) creating index... Traceback (most recent call last): File "train.py", line 326, in <module> train(opt) File "train.py", line 113, in train training_set = CocoDataset(root_dir=os.path.join(opt.data_path, params.project_name), set=params.train_set, File "E:\PorosityDetection\Yet-Another-EfficientDet-Pytorch-master\Yet-Another-EfficientDet-Pytorch-master\efficientdet\dataset.py", line 17, in __init__ self.coco = COCO(os.path.join(self.root_dir, 'annotations', 'instances_' + self.set_name + '.json')) File "C:\ProgramData\Anaconda3\envs\effdetpy\lib\site-packages\pycocotools\coco.py", line 89, in __init__ self.createIndex() File "C:\ProgramData\Anaconda3\envs\effdetpy\lib\site-packages\pycocotools\coco.py", line 111, in createIndex catToImgs[ann['category_id']].append(ann['image_id']) KeyError: 'category_id'

I did the annotations using (makesense.ai) that resulted the following:

For train

{"info":{"description":"PorosityDetection"},"images":[{"id":1,"width":1268,"height":690,"file_name":"04-%206%20a%208mm%20HD.JPG"},{"id":2,"width":739,"height":589,"file_name":"04_4-6mm_(HD)%20(2).JPG"},{"id":3,"width":1586,"height":700,"file_name":"0.jpg"},{"id":4,"width":1317,"height":651,"file_name":"01-%204%20a%206mm%20HD.JPG"},{"id":5,"width":763,"height":648,"file_name":"01_6-8mm_(HD)%20(3).JPG"},{"id":6,"width":1204,"height":699,"file_name":"01_6-8mm_(HD).JPG"},{"id":7,"width":1000,"height":1000,"file_name":"1%20(HD)%20(1)..jpg"},{"id":8,"width":1000,"height":1000,"file_name":"1%20(HD)%20(3)..jpg"},{"id":9,"width":1446,"height":719,"file_name":"1.jpg"},{"id":10,"width":1000,"height":1000,"file_name":"1-1_HD_8%20A%2010..jpg"},{"id":11,"width":1000,"height":1000,"file_name":"1-1HD_4%20a%206..jpg"},{"id":12,"width":848,"height":678,"file_name":"02.JPG"},{"id":13,"width":712,"height":678,"file_name":"02_4-6mm_(HD)%20(2).JPG"},{"id":14,"width":1115,"height":809,"file_name":"02_6-8mm_(HD).JPG"},{"id":15,"width":1000,"height":1000,"file_name":"2%20(HD)%20(1)..jpg"},{"id":16,"width":1000,"height":1000,"file_name":"2%20(HD)%20(2)..jpg"},{"id":17,"width":1000,"height":1000,"file_name":"2%20(HD)%20(4)..jpg"},{"id":18,"width":1416,"height":626,"file_name":"2.jpg"},{"id":19,"width":1000,"height":1000,"file_name":"2_HD_4%20a%206..jpg"},{"id":20,"width":1000,"height":1000,"file_name":"2-1_HD_4%20A%206..jpg"},{"id":21,"width":1000,"height":1000,"file_name":"2-1HD_8%20a%2010..jpg"},{"id":22,"width":1000,"height":1000,"file_name":"2-2%20(2)..jpg"},{"id":23,"width":787,"height":663,"file_name":"03_4-6mm_(HD)%20(2).JPG"},{"id":24,"width":1234,"height":720,"file_name":"03_4-6mm_(HD).JPG"},{"id":25,"width":1068,"height":823,"file_name":"03_6-8mm_(HD).JPG"},{"id":26,"width":1000,"height":1000,"file_name":"3%20(HD)%20(2)..jpg"},{"id":27,"width":1000,"height":1000,"file_name":"3%20(HD)%20(3)..jpg"},{"id":28,"width":1000,"height":1000,"file_name":"3%20(HD)%20(4)..jpg"},{"id":29,"width":822,"height":367,"file_name":"3.jpg"},{"id":30,"width":1000,"height":1000,"file_name":"3-1HD_8%20a%2010..jpg"}],"annotations":[{"id":0,"iscrowd":0,"image_id":1,"category_id": 1,"segmentation":[[707.8311965811967,501.4679487179488,957.7735042735043,519.7564102564104,962.5149572649573,657.2585470085471,650.2564102564103,649.130341880342]],"bbox":[650.2564102564103,501.4679487179488,312.258547008547,155.79059829059833],"area":40428.85575096792},{"id":1,"iscrowd":0,"image_id":1,"category_id": 1,"segmentation":[[514.1089743589744,518.4017094017095,562.8782051282052,624.7457264957266,296.6794871794872,610.5213675213677,291.9380341880342,550.91452991453]],"bbox":[291.9380341880342,518.4017094017095,270.940170940171,106.34401709401709],"area":20505.997564193898},{"id":2,"iscrowd":0,"image_id":2,"category_id": 1,"segmentation":[[365.4863713798978,474.61158432708686,302.2717206132879,494.1780238500852,317.824531516184,523.2768313458263,362.4761499148211,534.8160136286201,392.07666098807493,523.7785349233391]],"bbox":[302.2717206132879,474.61158432708686,89.80494037478701,60.20442930153325],"area":3307.800702834557},{"id":3,"iscrowd":0,"image_id":2,"category_id": 1,"segmentation":[[523.0212947189098,472.103066439523,606.8057921635434,501.70357751277686,604.2972742759796,542.8432708688246,518.5059625212947,495.1814310051107]],"bbox":[518.5059625212947,472.103066439523,88.29982964224871,70.74020442930157],"area":2858.127076869947},{"id":4,"iscrowd":0,"image_id":3,"category_id": 1,"segmentation":[[420.22222222222223,402.52777777777777,477.8333333333333,410.15277777777777,446.4861111111111,477.93055555555554,399.0416666666667,446.5833333333333]],"bbox":[399.0416666666667,402.52777777777777,78.79166666666663,75.40277777777777],"area":3448.9592978395012},{"id":5,"iscrowd":0,"image_id":4,"category_id": 1,"segmentation":[[738.7019230769231,464.79807692307696,968.7548076923077,483.08974358974365,972.9759615384617,616.7596153846155,726.0384615384615,614.6490384615386]],"bbox":[726.0384615384615,464.79807692307696,246.9375000000001,151.9615384615385],"area":33852.23833903479},{"id":6,"iscrowd":0,"image_id":4,"category_id": 1,"segmentation":[[324.32532051282055,474.64743589743597,561.4134615384615,476.05448717948724,541.011217948718,602.6891025641027,330.6570512820513,580.1762820512821]],"bbox":[324.32532051282055,474.64743589743597,237.088141025641,128.04166666666669],"area":26054.07996383959},{"id":7,"iscrowd":0,"image_id":5,"category_id": 1,"segmentation":[[375.42844974446336,479.1005110732538,467.60562180579217,507.25042589437817,452.15076660988075,540.3679727427598,335.68739352640546,536.5042589437819,337.34327086882456,487.3798977853492]],"bbox":[335.68739352640546,479.1005110732538,131.9182282793867,61.26746166950596],"area":6091.958638182772},{"id":8,"iscrowd":0,"image_id":5,"category_id": 1,"segmentation":[[191.6260647359455,478.54855195911415,251.2376490630324,482.41226575809196,257.30919931856903,527.6729131175468,198.80153321976152,517.7376490630323]],"bbox":[191.6260647359455,478.54855195911415,65.68313458262352,49.12436115843269],"area":2448.086299115707},{"id":9,"iscrowd":0,"image_id":6,"category_id": 1,"segmentation":[[333.8012820512821,551.4529914529915,556.3354700854701,586.1837606837607,548.6175213675214,674.2970085470087,344.09188034188037,635.707264957265]],"bbox":[333.8012820512821,551.4529914529915,222.534188034188,122.8440170940172],"area":18355.654056176503},{"id":10,"iscrowd":0,"image_id":7,"category_id": 1,"segmentation":[[512.7768313458263,78.36456558773425,443.7819420783646,79.21635434412266,448.0408858603066,100.51107325383305,521.2947189097104,102.21465076660988]],"bbox":[443.7819420783646,78.36456558773425,77.51277683134578,23.85008517887563],"area":1602.7268848909807},{"id":11,"iscrowd":0,"image_id":8,"category_id": 1,"segmentation":[[113.28790459965929,557.0698466780239,112.43611584327087,602.2146507666099,156.7291311754685,605.6218057921635,172.91311754684838,559.6252129471891]],"bbox":[112.43611584327087,557.0698466780239,60.477001703577514,48.55195911413966],"area":2393.207166053824},{"id":12,"iscrowd":0,"image_id":9,"category_id": 1,"segmentation":[[380.03846153846155,330.9198717948718,502.08333333333337,405.073717948718,437.97115384615387,486.17948717948724,314.3814102564103,452.96474358974365,305.8846153846154,371.85897435897436]],"bbox":[305.8846153846154,330.9198717948718,196.19871794871796,155.25961538461542],"area":19139.57016354373},{"id":13,"iscrowd":0,"image_id":10,"category_id": 1,"segmentation":[[177.17206132879045,532.3679727427598,321.1243611584327,519.5911413969336,339.86371379897787,574.9574105621806,198.46678023850086,606.473594548552,147.3594548551959,558.7734241908007]],"bbox":[147.3594548551959,519.5911413969336,192.50425894378196,86.88245315161839],"area":11065.272848108803},{"id":14,"iscrowd":0,"image_id":11,"category_id": 1,"segmentation":[[338.160136286201,766.6098807495741,166.95059625212946,804.9403747870529,206.1328790459966,859.4548551959114,350.93696763202723,853.4923339011925]],"bbox":[166.95059625212946,766.6098807495741,183.98637137989778,92.84497444633723],"area":11746.195972359663},{"id":15,"iscrowd":0,"image_id":12,"category_id": 1,"segmentation":[[434.3952299829642,251.21805792163545,426.88756388415675,280.09369676320273,454.03066439522996,276.0511073253833,463.27086882453153,261.0357751277683]],"bbox":[426.88756388415675,251.21805792163545,36.383304940374785,28.87563884156728],"area":638.859489681312},{"id":16,"iscrowd":0,"image_id":13,"category_id": 1,"segmentation":[[187.366269165247,554.9897785349234,291.8960817717206,580.977853492334,270.52810902896084,636.4190800681432,192.56388415672913,599.458262350937]],"bbox":[187.366269165247,554.9897785349234,104.52981260647363,81.42930153321981],"area":4812.708136831812},{"id":17,"iscrowd":0,"image_id":13,"segmentation":[[392.96081771720617,566.5400340715504,452.44463373083477,552.6797274275981,414.9063032367973,598.8807495741057,377.94548551959116,583.8654173764908,395.27086882453153,565.9625212947191,387.1856899488927,540.5519591141398]],"bbox":[377.94548551959116,540.5519591141398,74.49914821124361,58.328790459965944],"area":1499.510449285921},{"id":18,"iscrowd":0,"image_id":14,"category_id": 1,"segmentation":[[331.4761499148211,687.7189097103918,495.4812606473595,692.5425894378194,476.18654173764907,781.436115843271,362.4855195911414,737.3339011925043]],"bbox":[331.4761499148211,687.7189097103918,164.00511073253836,93.71720613287914],"area":9472.87877681975},{"id":19,"iscrowd":0,"image_id":14,"category_id": 1,"segmentation":[[658.7972742759796,698.0553662691652,782.8347529812606,695.9880749574106,758.0272572402043,771.7887563884157,662.2427597955707,773.8560477001704,656.0408858603066,698.7444633730835]],"bbox":[656.0408858603066,695.9880749574106,126.79386712095402,77.86797274275978],"area":8414.902240915464},{"id":20,"iscrowd":0,"image_id":15,"category_id": 1,"segmentation":[[798.9778534923339,560.4770017035775,862.8620102214651,569.8466780238501,836.4565587734241,603.0664395229983,784.4974446337309,601.3628620102214]],"bbox":[784.4974446337309,560.4770017035775,78.3645655877342,42.58943781942082],"area":2259.3442822772713},{"id":21,"iscrowd":0,"image_id":16,"category_id": 1,"segmentation":[[850.9369676320273,542.5894378194208,900.3407155025553,561.328790459966,888.4156729131175,580.0681431005111,849.2333901192504,572.4020442930154]],"bbox":[849.2333901192504,542.5894378194208,51.1073253833049,37.47870528109024],"area":1165.2238013286187},{"id":22,"iscrowd":0,"image_id":17,"category_id": 1,"segmentation":[[786.2010221465076,660.1362862010221,838.160136286201,660.9880749574105,817.717206132879,684.8381601362862,786.2010221465076,677.1720613287905]],"bbox":[786.2010221465076,660.1362862010221,51.95911413969338,24.701873935264075],"area":896.7724896900199},{"id":23,"iscrowd":0,"image_id":17,"category_id": 1,"segmentation":[[876.4906303236797,586.8824531516184,898.6371379897786,592.8449744463373,869.6763202725724,620.954003407155,855.1959114139694,599.6592844974447]],"bbox":[855.1959114139694,586.8824531516184,43.441226575809196,34.071550255536636],"area":716.837556483606},{"id":24,"iscrowd":0,"image_id":18,"category_id": 1,"segmentation":[[333.5769230769231,402.2564102564102,454.6025641025641,378.051282051282,430.3974358974359,542.1923076923076,312.3974358974359,543.7051282051282,311.64102564102564,428.73076923076917]],"bbox":[311.64102564102564,378.051282051282,142.96153846153845,165.6538461538462],"area":19240.191896778437},{"id":25,"iscrowd":0,"image_id":19,"category_id": 1,"segmentation":[[496.59284497444634,474.44633730834755,553.6626916524701,495.7410562180579,512.7768313458263,520.442930153322,479.557069846678,497.44463373083477]],"bbox":[479.557069846678,474.44633730834755,74.10562180579211,45.99659284497443],"area":1718.0883944870238},{"id":26,"iscrowd":0,"image_id":20,"category_id": 1,"segmentation":[[353.4923339011925,738.5008517887563,376.49063032367974,760.6473594548552,292.1635434412266,827.9386712095401,206.98466780238502,804.9403747870529,258.9437819420784,741.0562180579217]],"bbox":[206.98466780238502,738.5008517887563,169.50596252129472,89.43781942078374],"area":9175.593277398728},{"id":27,"iscrowd":0,"image_id":21,"category_id": 1,"segmentation":[[159.28449744463373,609.8807495741056,271.7206132879046,652.4701873935264,240.20442930153322,702.725724020443,132.02725724020442,641.396933560477]],"bbox":[132.02725724020442,609.8807495741056,139.69335604770018,92.84497444633735],"area":6036.889563483657},{"id":28,"iscrowd":0,"image_id":22,"category_id": 1,"segmentation":[[328.79045996592845,620.1022146507667,411.41396933560475,634.5826235093697,394.37819420783643,680.5792163543441,328.79045996592845,684.8381601362862]],"bbox":[328.79045996592845,620.1022146507667,82.6235093696763,64.73594548551955],"area":4146.484448688067},{"id":29,"iscrowd":0,"image_id":23,"category_id": 1,"segmentation":[[327.42589437819424,527.4633730834753,342.6737649063033,597.4906303236797,234.2444633730835,607.6558773424191,251.7512776831346,550.0528109028961]],"bbox":[234.2444633730835,527.4633730834753,108.42930153321979,80.19250425894381],"area":5855.8129206341855},{"id":30,"iscrowd":0,"image_id":23,"category_id": 1,"segmentation":[[455.0562180579217,548.9233390119251,502.4940374787053,524.0749574105622,512.6592844974447,564.1712095400342,495.7172061328791,586.7606473594549]],"bbox":[455.0562180579217,524.0749574105622,57.60306643952299,62.68568994889267],"area":1857.110104942698},{"id":31,"iscrowd":0,"image_id":24,"category_id": 1,"segmentation":[[376.3963675213675,605.8771367521367,568.2200854700855,592.034188034188,566.9017094017094,663.2264957264957,385.625,674.4326923076923]],"bbox":[376.3963675213675,592.034188034188,191.82371794871796,82.39850427350427],"area":13084.533165315223},{"id":32,"iscrowd":0,"image_id":25,"category_id": 1,"segmentation":[[289.3432708688245,695.4139693356047,387.4863713798978,679.2904599659283,383.9812606473594,776.732538330494,299.15758091993183,739.5783645655877,289.3432708688245,698.2180579216354]],"bbox":[289.3432708688245,679.2904599659283,98.14310051107327,97.44207836456565],"area":6457.90856765989},{"id":33,"iscrowd":0,"image_id":25,"category_id": 1,"segmentation":[[612.5144804088586,687.0017035775127,738.6984667802385,722.7538330494036,721.8739352640545,776.732538330494,604.8032367972743,736.774275979557]],"bbox":[604.8032367972743,687.0017035775127,133.89522998296422,89.73083475298131],"area":6773.899375741879},{"id":34,"iscrowd":0,"image_id":26,"category_id": 1,"segmentation":[[935.2640545144804,585.1788756388415,895.2299829642249,597.1039182282794,931.0051107325384,620.954003407155]],"bbox":[895.2299829642249,585.1788756388415,40.03407155025559,35.775127768313496],"area":690.7179694052611},{"id":35,"iscrowd":0,"image_id":27,"category_id": 1,"segmentation":[[913.1175468483816,562.1805792163543,856.8994889267461,600.511073253833,916.5247018739352,610.732538330494]],"bbox":[856.8994889267461,562.1805792163543,59.625212947189084,48.55195911413966],"area":1430.0473925396673},{"id":36,"iscrowd":0,"image_id":28,"category_id": 1,"segmentation":[[870.5281090289608,679.7274275979557,953.1516183986371,659.2844974446338,945.4855195911414,716.3543441226576,880.7495741056218,705.2810902896082]],"bbox":[870.5281090289608,659.2844974446338,82.6235093696763,57.06984667802385],"area":3049.824563440119},{"id":37,"iscrowd":0,"image_id":29,"category_id": 1,"segmentation":[[182.66666666666666,263.8557692307692,248.0929487179487,258.14743589743586,215.1602564102564,296.7884615384615]],"bbox":[182.66666666666666,258.14743589743586,65.42628205128204,38.641025641025635],"area":1170.073928542076},{"id":38,"iscrowd":0,"image_id":30,"category_id": 1,"segmentation":[[204.42930153321976,703.5775127768313,301.53321976149914,697.6149914821125,284.4974446337308,767.4616695059625,183.98637137989778,762.350936967632]],"bbox":[183.98637137989778,697.6149914821125,117.54684838160136,69.84667802385002],"area":6346.33411595355}],"categories":[{"id":1,"name":"Porosity"}]}

For Validation:

{"info":{"description":"PorosityDetection"},"images":[{"id":1,"width":1000,"height":1000,"file_name":"38%20(HD)%20(3)..jpg"},{"id":2,"width":1000,"height":1000,"file_name":"38%20(HD)%20(4)..jpg"},{"id":3,"width":1010,"height":557,"file_name":"38.jpg"},{"id":4,"width":1220,"height":769,"file_name":"38_2-4mm_(HD).JPG"},{"id":5,"width":1086,"height":516,"file_name":"38_6-8mm_(HD).JPG"},{"id":6,"width":1000,"height":1000,"file_name":"38-1_HD_4%20a%206..jpg"},{"id":7,"width":1172,"height":584,"file_name":"39_2-4mm_(HD)%20(2).JPG"}],"annotations":[{"id":0,"iscrowd":0,"image_id":1,"category_id": 1,"segmentation":[[926.7461669505963,570.6984667802385,890.1192504258944,579.2163543441227,916.5247018739352,596.252129471891]],"bbox":[890.1192504258944,570.6984667802385,36.62691652470187,25.55366269165245],"area":424.4432900231877},{"id":1,"iscrowd":0,"image_id":2,"category_id": 1,"segmentation":[[949.7444633730835,655.87734241908,924.190800681431,674.6166950596252,954.855195911414,683.9863713798978,966.7802385008517,673.7649063032368]],"bbox":[924.190800681431,655.87734241908,42.58943781942071,28.109028960817795],"area":600.7505028020532},{"id":2,"iscrowd":0,"image_id":3,"category_id": 1,"segmentation":[[530.8974358974359,345.4017094017094,622.0779914529915,364.2852564102564,608.0502136752136,412.84294871794873,534.1346153846154,407.9871794871795]],"bbox":[530.8974358974359,345.4017094017094,91.18055555555554,67.44123931623932],"area":4651.367515660387},{"id":3,"iscrowd":0,"image_id":4,"category_id": 1,"segmentation":[[462.61925042589434,642.5800681431006,539.2572402044293,646.5102214650767,539.2572402044293,688.4318568994889,464.5843270868824,686.4667802385009,443.62350936967624,661.5758091993185]],"bbox":[443.62350936967624,642.5800681431006,95.63373083475307,45.85178875638837],"area":3678.5328435668907},{"id":4,"iscrowd":0,"image_id":5,"category_id": 1,"segmentation":[[384.62500000000006,365.32371794871796,346.9166666666667,443.06089743589746,392.1666666666667,460.4647435897436,467.58333333333337,393.16987179487177]],"bbox":[346.9166666666667,365.32371794871796,120.66666666666669,95.14102564102564],"area":5928.305956196582},{"id":5,"iscrowd":0,"image_id":5,"category_id": 1,"segmentation":[[637.5608974358975,382.7275641025641,603.3333333333334,442.4807692307692,663.0865384615386,470.90705128205127,733.2820512820514,415.2147435897436]],"bbox":[603.3333333333334,382.7275641025641,129.948717948718,88.17948717948718],"area":6077.397045529251},{"id":6,"iscrowd":0,"image_id":6,"category_id": 1,"segmentation":[[314.31005110732536,742.7597955706984,199.31856899488926,764.9063032367973,214.65076660988075,838.160136286201,264.05451448040884,822.8279386712095,335.6047700170358,754.6848381601363]],"bbox":[199.31856899488926,742.7597955706984,136.2862010221465,95.40034071550258],"area":7126.294007876506},{"id":7,"iscrowd":0,"image_id":7,"category_id": 1,"segmentation":[[471.4294871794872,412.20512820512823,443.2564102564103,479.8205128205129,495.84615384615387,492.9679487179488,511.4978632478632,421.59615384615387]],"bbox":[443.2564102564103,412.20512820512823,68.24145299145295,80.76282051282055],"area":3466.508492220033},{"id":8,"iscrowd":0,"image_id":7,"category_id": 1,"segmentation":[[801.9935897435897,430.98717948717956,863.3482905982906,487.95940170940173,818.2713675213676,535.5405982905984,774.4465811965812,508.61965811965814,772.5683760683761,451.64743589743597]],"bbox":[772.5683760683761,430.98717948717956,90.77991452991455,104.55341880341882],"area":5673.252365037617}],"categories":[{"id":1,"name":"Porosity"}]}

My Yaml

`project_name: PorosityDetection # also the folder name of the dataset that under data_path folder train_set: train_porosity val_set: val_porosity num_gpus: 2

mean and std in RGB order, actually this part should remain unchanged as long as your dataset is similar to coco. mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225]

this is coco anchors, change it if necessary anchors_scales: '[2 0, 2 (1.0 / 3.0), 2 ** (2.0 / 3.0)]' anchors_ratios: '[(1.0, 1.0), (1.4, 0.7), (0.7, 1.4)]'

must match your dataset's category_id. category_id is one_indexed, obj_list: [ 'Porosity']`

I double checked and Category_ID is an integer, and I cant figure out what is generating the error. Also, if necessary, I can relabel my images with another software and would be really pleased if you indicate one that works flawlessly.

zylo117 commented 3 years ago

I think some of your annotations are missing category_id but most of them aren't. So when the program tried to get their category_id, it crashed.

valterfsj commented 3 years ago

Just found out that img 17 had no Category_ID.

Now I managed to run but still got tons of errors, as following:

(effdetpy) E:\PorosityDetection\Yet-Another-EfficientDet-Pytorch-master\Yet-Another-EfficientDet-Pytorch-master>python train.py -c 2 -p PorosityDetection --batch_size 8 --lr 1e-3 --num_epochs 10 --load_weights E:/PorosityDetection/Yet-Another-EfficientDet-Pytorch-master/Yet-Another-EfficientDet-Pytorch-master/weights/efficientdet-d0.pth --head_only True loading annotations into memory... Done (t=0.00s) creating index... index created! loading annotations into memory... Done (t=0.00s) creating index... index created! [Warning] Ignoring Error(s) in loading state_dict for EfficientDetBackbone: size mismatch for bifpn.0.conv6_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv6_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv6_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv5_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv5_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv4_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv4_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv3_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv3_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv3_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv3_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv3_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv3_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv3_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv4_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv4_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv4_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv5_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv5_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv5_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv6_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv6_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv6_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv7_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.0.conv7_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.0.conv7_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv7_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv7_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv7_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.conv7_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel.0.conv.weight: copying a param with shape torch.Size([64, 320, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 352, 1, 1]). size mismatch for bifpn.0.p5_down_channel.0.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel.0.conv.weight: copying a param with shape torch.Size([64, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 120, 1, 1]). size mismatch for bifpn.0.p4_down_channel.0.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p3_down_channel.0.conv.weight: copying a param with shape torch.Size([64, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 48, 1, 1]). size mismatch for bifpn.0.p3_down_channel.0.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p3_down_channel.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p3_down_channel.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p3_down_channel.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p3_down_channel.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_to_p6.0.conv.weight: copying a param with shape torch.Size([64, 320, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 352, 1, 1]). size mismatch for bifpn.0.p5_to_p6.0.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_to_p6.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_to_p6.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_to_p6.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_to_p6.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel_2.0.conv.weight: copying a param with shape torch.Size([64, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 120, 1, 1]). size mismatch for bifpn.0.p4_down_channel_2.0.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel_2.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel_2.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel_2.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p4_down_channel_2.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel_2.0.conv.weight: copying a param with shape torch.Size([64, 320, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 352, 1, 1]). size mismatch for bifpn.0.p5_down_channel_2.0.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel_2.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel_2.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel_2.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.0.p5_down_channel_2.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv6_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv6_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv5_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv5_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv4_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv4_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv3_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv3_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv3_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv3_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv3_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv3_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv3_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv4_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv4_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv4_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv5_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv5_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv5_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv6_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv6_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv6_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv7_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.1.conv7_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.1.conv7_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv7_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv7_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv7_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.1.conv7_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv6_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv6_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv5_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv5_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv4_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv4_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv3_up.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv3_up.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv3_up.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv3_up.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv3_up.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv3_up.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv3_up.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv4_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv4_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv4_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv5_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv5_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv5_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv6_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv6_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv6_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv7_down.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for bifpn.2.conv7_down.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for bifpn.2.conv7_down.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv7_down.bn.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv7_down.bn.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv7_down.bn.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for bifpn.2.conv7_down.bn.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.conv_list.0.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for regressor.conv_list.0.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for regressor.conv_list.0.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.conv_list.1.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for regressor.conv_list.1.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for regressor.conv_list.1.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.conv_list.2.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for regressor.conv_list.2.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for regressor.conv_list.2.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.0.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.1.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.2.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.3.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.bn_list.4.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for regressor.header.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for regressor.header.pointwise_conv.conv.weight: copying a param with shape torch.Size([36, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([36, 112, 1, 1]). size mismatch for classifier.conv_list.0.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for classifier.conv_list.0.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for classifier.conv_list.0.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.conv_list.1.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for classifier.conv_list.1.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for classifier.conv_list.1.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.conv_list.2.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for classifier.conv_list.2.pointwise_conv.conv.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([112, 112, 1, 1]). size mismatch for classifier.conv_list.2.pointwise_conv.conv.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.0.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.1.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.2.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.3.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.0.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.0.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.0.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.0.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.1.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.1.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.2.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.2.running_mean: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.bn_list.4.2.running_var: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([112]). size mismatch for classifier.header.depthwise_conv.conv.weight: copying a param with shape torch.Size([64, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([112, 1, 3, 3]). size mismatch for classifier.header.pointwise_conv.conv.weight: copying a param with shape torch.Size([810, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([9, 112, 1, 1]). size mismatch for classifier.header.pointwise_conv.conv.bias: copying a param with shape torch.Size([810]) from checkpoint, the shape in current model is torch.Size([9]). size mismatch for backbone_net.model._blocks.1._depthwise_conv.conv.weight: copying a param with shape torch.Size([96, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([16, 1, 3, 3]). size mismatch for backbone_net.model._blocks.1._bn1.weight: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._bn1.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._bn1.running_mean: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._bn1.running_var: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._se_reduce.conv.weight: copying a param with shape torch.Size([4, 96, 1, 1]) from checkpoint, the shape in current model is torch.Size([4, 16, 1, 1]). size mismatch for backbone_net.model._blocks.1._se_expand.conv.weight: copying a param with shape torch.Size([96, 4, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 4, 1, 1]). size mismatch for backbone_net.model._blocks.1._se_expand.conv.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._project_conv.conv.weight: copying a param with shape torch.Size([24, 96, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 16, 1, 1]). size mismatch for backbone_net.model._blocks.1._bn2.weight: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._bn2.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._bn2.running_mean: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.1._bn2.running_var: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([16]). size mismatch for backbone_net.model._blocks.2._expand_conv.conv.weight: copying a param with shape torch.Size([144, 24, 1, 1]) from checkpoint, the shape in current model is torch.Size([96, 16, 1, 1]). size mismatch for backbone_net.model._blocks.2._bn0.weight: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._bn0.bias: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._bn0.running_mean: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._bn0.running_var: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._depthwise_conv.conv.weight: copying a param with shape torch.Size([144, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([96, 1, 3, 3]). size mismatch for backbone_net.model._blocks.2._bn1.weight: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._bn1.bias: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._bn1.running_mean: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._bn1.running_var: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._se_reduce.conv.weight: copying a param with shape torch.Size([6, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([4, 96, 1, 1]). size mismatch for backbone_net.model._blocks.2._se_reduce.conv.bias: copying a param with shape torch.Size([6]) from checkpoint, the shape in current model is torch.Size([4]). size mismatch for backbone_net.model._blocks.2._se_expand.conv.weight: copying a param with shape torch.Size([144, 6, 1, 1]) from checkpoint, the shape in current model is torch.Size([96, 4, 1, 1]). size mismatch for backbone_net.model._blocks.2._se_expand.conv.bias: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([96]). size mismatch for backbone_net.model._blocks.2._project_conv.conv.weight: copying a param with shape torch.Size([24, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([24, 96, 1, 1]). size mismatch for backbone_net.model._blocks.3._depthwise_conv.conv.weight: copying a param with shape torch.Size([144, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([144, 1, 3, 3]). size mismatch for backbone_net.model._blocks.3._project_conv.conv.weight: copying a param with shape torch.Size([40, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([24, 144, 1, 1]). size mismatch for backbone_net.model._blocks.3._bn2.weight: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.3._bn2.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.3._bn2.running_mean: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.3._bn2.running_var: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.4._expand_conv.conv.weight: copying a param with shape torch.Size([240, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([144, 24, 1, 1]). size mismatch for backbone_net.model._blocks.4._bn0.weight: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._bn0.bias: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._bn0.running_mean: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._bn0.running_var: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._depthwise_conv.conv.weight: copying a param with shape torch.Size([240, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([144, 1, 3, 3]). size mismatch for backbone_net.model._blocks.4._bn1.weight: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._bn1.bias: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._bn1.running_mean: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._bn1.running_var: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._se_reduce.conv.weight: copying a param with shape torch.Size([10, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([6, 144, 1, 1]). size mismatch for backbone_net.model._blocks.4._se_reduce.conv.bias: copying a param with shape torch.Size([10]) from checkpoint, the shape in current model is torch.Size([6]). size mismatch for backbone_net.model._blocks.4._se_expand.conv.weight: copying a param with shape torch.Size([240, 10, 1, 1]) from checkpoint, the shape in current model is torch.Size([144, 6, 1, 1]). size mismatch for backbone_net.model._blocks.4._se_expand.conv.bias: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.4._project_conv.conv.weight: copying a param with shape torch.Size([40, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([24, 144, 1, 1]). size mismatch for backbone_net.model._blocks.4._bn2.weight: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.4._bn2.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.4._bn2.running_mean: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.4._bn2.running_var: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([24]). size mismatch for backbone_net.model._blocks.5._expand_conv.conv.weight: copying a param with shape torch.Size([240, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([144, 24, 1, 1]). size mismatch for backbone_net.model._blocks.5._bn0.weight: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._bn0.bias: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._bn0.running_mean: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._bn0.running_var: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._depthwise_conv.conv.weight: copying a param with shape torch.Size([240, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([144, 1, 5, 5]). size mismatch for backbone_net.model._blocks.5._bn1.weight: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._bn1.bias: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._bn1.running_mean: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._bn1.running_var: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._se_reduce.conv.weight: copying a param with shape torch.Size([10, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([6, 144, 1, 1]). size mismatch for backbone_net.model._blocks.5._se_reduce.conv.bias: copying a param with shape torch.Size([10]) from checkpoint, the shape in current model is torch.Size([6]). size mismatch for backbone_net.model._blocks.5._se_expand.conv.weight: copying a param with shape torch.Size([240, 10, 1, 1]) from checkpoint, the shape in current model is torch.Size([144, 6, 1, 1]). size mismatch for backbone_net.model._blocks.5._se_expand.conv.bias: copying a param with shape torch.Size([240]) from checkpoint, the shape in current model is torch.Size([144]). size mismatch for backbone_net.model._blocks.5._project_conv.conv.weight: copying a param with shape torch.Size([80, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 144, 1, 1]). size mismatch for backbone_net.model._blocks.5._bn2.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.5._bn2.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.5._bn2.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.5._bn2.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.6._expand_conv.conv.weight: copying a param with shape torch.Size([480, 80, 1, 1]) from checkpoint, the shape in current model is torch.Size([288, 48, 1, 1]). size mismatch for backbone_net.model._blocks.6._bn0.weight: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._bn0.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._bn0.running_mean: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._bn0.running_var: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._depthwise_conv.conv.weight: copying a param with shape torch.Size([480, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([288, 1, 5, 5]). size mismatch for backbone_net.model._blocks.6._bn1.weight: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._bn1.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._bn1.running_mean: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._bn1.running_var: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._se_reduce.conv.weight: copying a param with shape torch.Size([20, 480, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 288, 1, 1]). size mismatch for backbone_net.model._blocks.6._se_reduce.conv.bias: copying a param with shape torch.Size([20]) from checkpoint, the shape in current model is torch.Size([12]). size mismatch for backbone_net.model._blocks.6._se_expand.conv.weight: copying a param with shape torch.Size([480, 20, 1, 1]) from checkpoint, the shape in current model is torch.Size([288, 12, 1, 1]). size mismatch for backbone_net.model._blocks.6._se_expand.conv.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.6._project_conv.conv.weight: copying a param with shape torch.Size([80, 480, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 288, 1, 1]). size mismatch for backbone_net.model._blocks.6._bn2.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.6._bn2.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.6._bn2.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.6._bn2.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.7._expand_conv.conv.weight: copying a param with shape torch.Size([480, 80, 1, 1]) from checkpoint, the shape in current model is torch.Size([288, 48, 1, 1]). size mismatch for backbone_net.model._blocks.7._bn0.weight: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._bn0.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._bn0.running_mean: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._bn0.running_var: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._depthwise_conv.conv.weight: copying a param with shape torch.Size([480, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([288, 1, 5, 5]). size mismatch for backbone_net.model._blocks.7._bn1.weight: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._bn1.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._bn1.running_mean: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._bn1.running_var: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._se_reduce.conv.weight: copying a param with shape torch.Size([20, 480, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 288, 1, 1]). size mismatch for backbone_net.model._blocks.7._se_reduce.conv.bias: copying a param with shape torch.Size([20]) from checkpoint, the shape in current model is torch.Size([12]). size mismatch for backbone_net.model._blocks.7._se_expand.conv.weight: copying a param with shape torch.Size([480, 20, 1, 1]) from checkpoint, the shape in current model is torch.Size([288, 12, 1, 1]). size mismatch for backbone_net.model._blocks.7._se_expand.conv.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.7._project_conv.conv.weight: copying a param with shape torch.Size([80, 480, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 288, 1, 1]). size mismatch for backbone_net.model._blocks.7._bn2.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.7._bn2.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.7._bn2.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.7._bn2.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for backbone_net.model._blocks.8._expand_conv.conv.weight: copying a param with shape torch.Size([480, 80, 1, 1]) from checkpoint, the shape in current model is torch.Size([288, 48, 1, 1]). size mismatch for backbone_net.model._blocks.8._bn0.weight: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._bn0.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._bn0.running_mean: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._bn0.running_var: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._depthwise_conv.conv.weight: copying a param with shape torch.Size([480, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([288, 1, 3, 3]). size mismatch for backbone_net.model._blocks.8._bn1.weight: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._bn1.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._bn1.running_mean: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._bn1.running_var: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._se_reduce.conv.weight: copying a param with shape torch.Size([20, 480, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 288, 1, 1]). size mismatch for backbone_net.model._blocks.8._se_reduce.conv.bias: copying a param with shape torch.Size([20]) from checkpoint, the shape in current model is torch.Size([12]). size mismatch for backbone_net.model._blocks.8._se_expand.conv.weight: copying a param with shape torch.Size([480, 20, 1, 1]) from checkpoint, the shape in current model is torch.Size([288, 12, 1, 1]). size mismatch for backbone_net.model._blocks.8._se_expand.conv.bias: copying a param with shape torch.Size([480]) from checkpoint, the shape in current model is torch.Size([288]). size mismatch for backbone_net.model._blocks.8._project_conv.conv.weight: copying a param with shape torch.Size([112, 480, 1, 1]) from checkpoint, the shape in current model is torch.Size([88, 288, 1, 1]). size mismatch for backbone_net.model._blocks.8._bn2.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.8._bn2.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.8._bn2.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.8._bn2.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.9._expand_conv.conv.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 88, 1, 1]). size mismatch for backbone_net.model._blocks.9._bn0.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._bn0.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._bn0.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._bn0.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._depthwise_conv.conv.weight: copying a param with shape torch.Size([672, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([528, 1, 3, 3]). size mismatch for backbone_net.model._blocks.9._bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._se_reduce.conv.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([22, 528, 1, 1]). size mismatch for backbone_net.model._blocks.9._se_reduce.conv.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([22]). size mismatch for backbone_net.model._blocks.9._se_expand.conv.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 22, 1, 1]). size mismatch for backbone_net.model._blocks.9._se_expand.conv.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.9._project_conv.conv.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([88, 528, 1, 1]). size mismatch for backbone_net.model._blocks.9._bn2.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.9._bn2.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.9._bn2.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.9._bn2.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.10._expand_conv.conv.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 88, 1, 1]). size mismatch for backbone_net.model._blocks.10._bn0.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._bn0.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._bn0.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._bn0.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._depthwise_conv.conv.weight: copying a param with shape torch.Size([672, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([528, 1, 3, 3]). size mismatch for backbone_net.model._blocks.10._bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._se_reduce.conv.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([22, 528, 1, 1]). size mismatch for backbone_net.model._blocks.10._se_reduce.conv.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([22]). size mismatch for backbone_net.model._blocks.10._se_expand.conv.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 22, 1, 1]). size mismatch for backbone_net.model._blocks.10._se_expand.conv.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.10._project_conv.conv.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([88, 528, 1, 1]). size mismatch for backbone_net.model._blocks.10._bn2.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.10._bn2.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.10._bn2.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.10._bn2.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.11._expand_conv.conv.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 88, 1, 1]). size mismatch for backbone_net.model._blocks.11._bn0.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._bn0.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._bn0.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._bn0.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._depthwise_conv.conv.weight: copying a param with shape torch.Size([672, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([528, 1, 3, 3]). size mismatch for backbone_net.model._blocks.11._bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._se_reduce.conv.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([22, 528, 1, 1]). size mismatch for backbone_net.model._blocks.11._se_reduce.conv.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([22]). size mismatch for backbone_net.model._blocks.11._se_expand.conv.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 22, 1, 1]). size mismatch for backbone_net.model._blocks.11._se_expand.conv.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.11._project_conv.conv.weight: copying a param with shape torch.Size([192, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([88, 528, 1, 1]). size mismatch for backbone_net.model._blocks.11._bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.11._bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.11._bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.11._bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([88]). size mismatch for backbone_net.model._blocks.12._expand_conv.conv.weight: copying a param with shape torch.Size([1152, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 88, 1, 1]). size mismatch for backbone_net.model._blocks.12._bn0.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._bn0.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._bn0.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._bn0.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._depthwise_conv.conv.weight: copying a param with shape torch.Size([1152, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([528, 1, 5, 5]). size mismatch for backbone_net.model._blocks.12._bn1.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._bn1.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._bn1.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._bn1.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._se_reduce.conv.weight: copying a param with shape torch.Size([48, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([22, 528, 1, 1]). size mismatch for backbone_net.model._blocks.12._se_reduce.conv.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([22]). size mismatch for backbone_net.model._blocks.12._se_expand.conv.weight: copying a param with shape torch.Size([1152, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([528, 22, 1, 1]). size mismatch for backbone_net.model._blocks.12._se_expand.conv.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([528]). size mismatch for backbone_net.model._blocks.12._project_conv.conv.weight: copying a param with shape torch.Size([192, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([120, 528, 1, 1]). size mismatch for backbone_net.model._blocks.12._bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.12._bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.12._bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.12._bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.13._expand_conv.conv.weight: copying a param with shape torch.Size([1152, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([720, 120, 1, 1]). size mismatch for backbone_net.model._blocks.13._bn0.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._bn0.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._bn0.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._bn0.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._depthwise_conv.conv.weight: copying a param with shape torch.Size([1152, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([720, 1, 5, 5]). size mismatch for backbone_net.model._blocks.13._bn1.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._bn1.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._bn1.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._bn1.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._se_reduce.conv.weight: copying a param with shape torch.Size([48, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([30, 720, 1, 1]). size mismatch for backbone_net.model._blocks.13._se_reduce.conv.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([30]). size mismatch for backbone_net.model._blocks.13._se_expand.conv.weight: copying a param with shape torch.Size([1152, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([720, 30, 1, 1]). size mismatch for backbone_net.model._blocks.13._se_expand.conv.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.13._project_conv.conv.weight: copying a param with shape torch.Size([192, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([120, 720, 1, 1]). size mismatch for backbone_net.model._blocks.13._bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.13._bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.13._bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.13._bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.14._expand_conv.conv.weight: copying a param with shape torch.Size([1152, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([720, 120, 1, 1]). size mismatch for backbone_net.model._blocks.14._bn0.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._bn0.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._bn0.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._bn0.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._depthwise_conv.conv.weight: copying a param with shape torch.Size([1152, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([720, 1, 5, 5]). size mismatch for backbone_net.model._blocks.14._bn1.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._bn1.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._bn1.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._bn1.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._se_reduce.conv.weight: copying a param with shape torch.Size([48, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([30, 720, 1, 1]). size mismatch for backbone_net.model._blocks.14._se_reduce.conv.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([30]). size mismatch for backbone_net.model._blocks.14._se_expand.conv.weight: copying a param with shape torch.Size([1152, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([720, 30, 1, 1]). size mismatch for backbone_net.model._blocks.14._se_expand.conv.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.14._project_conv.conv.weight: copying a param with shape torch.Size([192, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([120, 720, 1, 1]). size mismatch for backbone_net.model._blocks.14._bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.14._bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.14._bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.14._bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.15._expand_conv.conv.weight: copying a param with shape torch.Size([1152, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([720, 120, 1, 1]). size mismatch for backbone_net.model._blocks.15._bn0.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._bn0.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._bn0.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._bn0.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._depthwise_conv.conv.weight: copying a param with shape torch.Size([1152, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([720, 1, 5, 5]). size mismatch for backbone_net.model._blocks.15._bn1.weight: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._bn1.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._bn1.running_mean: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._bn1.running_var: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._se_reduce.conv.weight: copying a param with shape torch.Size([48, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([30, 720, 1, 1]). size mismatch for backbone_net.model._blocks.15._se_reduce.conv.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([30]). size mismatch for backbone_net.model._blocks.15._se_expand.conv.weight: copying a param with shape torch.Size([1152, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([720, 30, 1, 1]). size mismatch for backbone_net.model._blocks.15._se_expand.conv.bias: copying a param with shape torch.Size([1152]) from checkpoint, the shape in current model is torch.Size([720]). size mismatch for backbone_net.model._blocks.15._project_conv.conv.weight: copying a param with shape torch.Size([320, 1152, 1, 1]) from checkpoint, the shape in current model is torch.Size([120, 720, 1, 1]). size mismatch for backbone_net.model._blocks.15._bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.15._bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.15._bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([120]). size mismatch for backbone_net.model._blocks.15._bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([120]). [Warning] Don't panic if you see this, this might be because you load a pretrained weights with different number of classes. The rest of the weights should be loaded already. [Info] loaded weights: efficientdet-d0.pth, resuming checkpoint from step: 0 [Info] freezed backbone 0%| | 0/3 [00:00<?, ?it/s]C:\ProgramData\Anaconda3\envs\effdetpy\lib\site-packages\torch\cuda\nccl.py:16: UserWarning: PyTorch is not compiled with NCCL support warnings.warn('PyTorch is not compiled with NCCL support') Step: 2. Epoch: 0/10. Iteration: 3/3. Cls loss: 592.75928. Reg loss: 9.23222. Total loss: 601.99152: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [00:08<00:00, 2.94s/it] C:\ProgramData\Anaconda3\envs\effdetpy\lib\site-packages\numpy\core\fromnumeric.py:3372: RuntimeWarning: Mean of empty slice. return _methods._mean(a, axis=axis, dtype=dtype, C:\ProgramData\Anaconda3\envs\effdetpy\lib\site-packages\numpy\core\_methods.py:170: RuntimeWarning: invalid value encountered in double_scalars ret = ret.dtype.type(ret / rcount) Val. Epoch: 0/10. Classification loss: nan. Regression loss: nan. Total loss: nan WARNING:root:NaN or Inf found in input tensor. WARNING:root:NaN or Inf found in input tensor. WARNING:root:NaN or Inf found in input tensor. WARNING:root:NaN or Inf found in input tensor. WARNING:root:NaN or Inf found in input tensor. WARNING:root:NaN or Inf found in input tensor.

The first chunk of errors, is due the pretreined weights having more classes than my dataset, but after it starts training, the "WARNING:root:NaN or Inf found in input tensor." I couldnt find out what does it mean and after I let it train, there's no resulting weight saved.

zylo117 commented 3 years ago

-c 2 means running on a D2 model while you loaded a D0 pretrained weights