carlosgomes98
commited on
Commit
•
3b8de5a
1
Parent(s):
dd316eb
Pretrained in EncoderDecoder
Browse files
multi_temporal_crop_classification_Prithvi_100M.py
CHANGED
@@ -185,9 +185,9 @@ norm_cfg = dict(type='BN', requires_grad=True)
|
|
185 |
model = dict(
|
186 |
type='TemporalEncoderDecoder',
|
187 |
frozen_backbone=False,
|
|
|
188 |
backbone=dict(
|
189 |
type='TemporalViTEncoder',
|
190 |
-
pretrained=pretrained_weights_path,
|
191 |
img_size=img_size,
|
192 |
patch_size=patch_size,
|
193 |
num_frames=num_frames,
|
|
|
185 |
model = dict(
|
186 |
type='TemporalEncoderDecoder',
|
187 |
frozen_backbone=False,
|
188 |
+
pretrained=pretrained_weights_path,
|
189 |
backbone=dict(
|
190 |
type='TemporalViTEncoder',
|
|
|
191 |
img_size=img_size,
|
192 |
patch_size=patch_size,
|
193 |
num_frames=num_frames,
|