Paolo-Fraccaro
commited on
Commit
•
fed16ce
1
Parent(s):
67b68ed
fix bug
Browse files
burn_scars_Prithvi_100M.py
CHANGED
@@ -218,6 +218,6 @@ model = dict(
|
|
218 |
align_corners=False,
|
219 |
loss_decode=loss_func),
|
220 |
train_cfg=dict(),
|
221 |
-
test_cfg=dict(mode='slide', stride=(tile_size/2, tile_size/2), crop_size=(tile_size, tile_size)))
|
222 |
gpu_ids = range(0, 1)
|
223 |
auto_resume = False
|
|
|
218 |
align_corners=False,
|
219 |
loss_decode=loss_func),
|
220 |
train_cfg=dict(),
|
221 |
+
test_cfg=dict(mode='slide', stride=(int(tile_size/2), int(tile_size/2)), crop_size=(tile_size, tile_size)))
|
222 |
gpu_ids = range(0, 1)
|
223 |
auto_resume = False
|