@@ -104,90 +104,94 @@ def _get_segmentor_cfg(fname):
104104
105105def test_pspnet_forward ():
106106 _test_encoder_decoder_forward (
107- 'pspnet/pspnet_r18-d8_512x1024_80k_cityscapes .py' )
107+ 'pspnet/pspnet_r18-d8_4xb2-80k_cityscapes-512x1024 .py' )
108108
109109
110110def test_fcn_forward ():
111- _test_encoder_decoder_forward ('fcn/fcn_r18-d8_512x1024_80k_cityscapes.py' )
111+ _test_encoder_decoder_forward (
112+ 'fcn/fcn_r18-d8_4xb2-80k_cityscapes-512x1024.py' )
112113
113114
114115def test_deeplabv3_forward ():
115116 _test_encoder_decoder_forward (
116- 'deeplabv3/deeplabv3_r18-d8_512x1024_80k_cityscapes .py' )
117+ 'deeplabv3/deeplabv3_r18-d8_4xb2-80k_cityscapes-512x1024 .py' )
117118
118119
119120def test_deeplabv3plus_forward ():
120121 _test_encoder_decoder_forward (
121- 'deeplabv3plus/deeplabv3plus_r18-d8_512x1024_80k_cityscapes .py' )
122+ 'deeplabv3plus/deeplabv3plus_r18-d8_4xb2-80k_cityscapes-512x1024 .py' )
122123
123124
124125def test_gcnet_forward ():
125126 _test_encoder_decoder_forward (
126- 'gcnet/gcnet_r50-d8_512x1024_40k_cityscapes .py' )
127+ 'gcnet/gcnet_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
127128
128129
129130def test_ann_forward ():
130- _test_encoder_decoder_forward ('ann/ann_r50-d8_512x1024_40k_cityscapes.py' )
131+ _test_encoder_decoder_forward (
132+ 'ann/ann_r50-d8_4xb2-40k_cityscapes-512x1024.py' )
131133
132134
133135def test_ccnet_forward ():
134136 if not torch .cuda .is_available ():
135137 pytest .skip ('CCNet requires CUDA' )
136138 _test_encoder_decoder_forward (
137- 'ccnet/ccnet_r50-d8_512x1024_40k_cityscapes .py' )
139+ 'ccnet/ccnet_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
138140
139141
140142def test_danet_forward ():
141143 _test_encoder_decoder_forward (
142- 'danet/danet_r50-d8_512x1024_40k_cityscapes .py' )
144+ 'danet/danet_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
143145
144146
145147def test_nonlocal_net_forward ():
146148 _test_encoder_decoder_forward (
147- 'nonlocal_net/nonlocal_r50-d8_512x1024_40k_cityscapes .py' )
149+ 'nonlocal_net/nonlocal_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
148150
149151
150152def test_upernet_forward ():
151153 _test_encoder_decoder_forward (
152- 'upernet/upernet_r50_512x1024_40k_cityscapes .py' )
154+ 'upernet/upernet_r50_4xb2-40k_cityscapes-512x1024 .py' )
153155
154156
155157def test_hrnet_forward ():
156- _test_encoder_decoder_forward ('hrnet/fcn_hr18s_512x1024_40k_cityscapes.py' )
158+ _test_encoder_decoder_forward (
159+ 'hrnet/fcn_hr18s_4xb2-40k_cityscapes-512x1024.py' )
157160
158161
159162def test_ocrnet_forward ():
160163 _test_encoder_decoder_forward (
161- 'ocrnet/ocrnet_hr18s_512x1024_40k_cityscapes .py' )
164+ 'ocrnet/ocrnet_hr18s_4xb2-40k_cityscapes-512x1024 .py' )
162165
163166
164167def test_psanet_forward ():
165168 _test_encoder_decoder_forward (
166- 'psanet/psanet_r50-d8_512x1024_40k_cityscapes .py' )
169+ 'psanet/psanet_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
167170
168171
169172def test_sem_fpn_forward ():
170- _test_encoder_decoder_forward ('sem_fpn/fpn_r50_512x1024_80k_cityscapes.py' )
173+ _test_encoder_decoder_forward (
174+ 'sem_fpn/fpn_r50_4xb2-80k_cityscapes-512x1024.py' )
171175
172176
173177def test_mobilenet_v2_forward ():
174178 _test_encoder_decoder_forward (
175- 'mobilenet_v2/pspnet_m -v2-d8_512x1024_80k_cityscapes .py' )
179+ 'mobilenet_v2/mobilenet -v2-d8_pspnet_4xb2-80k_cityscapes-512x1024 .py' )
176180
177181
178182def test_dnlnet_forward ():
179183 _test_encoder_decoder_forward (
180- 'dnlnet/dnl_r50-d8_512x1024_40k_cityscapes .py' )
184+ 'dnlnet/dnl_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
181185
182186
183187def test_emanet_forward ():
184188 _test_encoder_decoder_forward (
185- 'emanet/emanet_r50-d8_512x1024_80k_cityscapes .py' )
189+ 'emanet/emanet_r50-d8_4xb2-80k_cityscapes-512x1024 .py' )
186190
187191
188192def test_isanet_forward ():
189193 _test_encoder_decoder_forward (
190- 'isanet/isanet_r50-d8_512x1024_40k_cityscapes .py' )
194+ 'isanet/isanet_r50-d8_4xb2-40k_cityscapes-512x1024 .py' )
191195
192196
193197def get_world_size (process_group ):
0 commit comments