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,