mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add TPU trained darknet53 weights. Add mising pretrain_cfg for some csp/darknet models.

edgenext_csp_and_more
Ross Wightman 2022-07-02 15:18:52 -07:00
parent c170ba3173
commit 377e9bfa21
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
timm/models/cspnet.py
View File

@ -45,14 +45,18 @@ default_cfgs = {
'cspresnet50w': _cfg(url=''),
'cspresnext50': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspresnext50_ra_224-648b4713.pth',
input_size=(3, 224, 224), pool_size=(7, 7), crop_pct=0.875 # FIXME I trained this at 224x224, not 256 like ref impl
),
'cspdarknet53': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspdarknet53_ra_256-d05c7c21.pth'),
'darknet17': _cfg(url=''),
'darknet21': _cfg(url=''),
'darknet53': _cfg(url=''),
'sedarknet21': _cfg(url=''),
'darknet53': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/darknet53_256_c2ns-3aeff817.pth',
test_input_size=(3, 288, 288), test_crop_pct=1.0, interpolation='bicubic'
),
'darknetaa53': _cfg(url=''),
'cs2darknet_m': _cfg(
url=''),