Skip to content

Commit

Permalink
added regression (segmentors) config
Browse files Browse the repository at this point in the history
  • Loading branch information
RituYadav92 authored Sep 17, 2024
1 parent 3f194c0 commit c369971
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 3 deletions.
24 changes: 24 additions & 0 deletions configs/segmentors/reg_upernet.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
segmentor_name: UPerNet_regress
task_name: regression
# time_frames: 2
#task_model_args:
#num_frames: 1
#mt_strategy: "ltae" #activated only when if num_frames > 1
#num_classes - task parameter passed from the dataset config
#wave_list - task parameter passed from the dataset config

channels: 512

loss:
loss_name: MSELoss # WeightedCrossEntropy
ignore_index: -1

optimizer:
optimizer_name: AdamW
lr: 0.0001
weight_decay: 0.05

scheduler:
scheduler_name: MultiStepLR
lr_milestones: [0.6, 0.9]

5 changes: 2 additions & 3 deletions configs/segmentors/reg_upernet_mt.yaml
Original file line number Diff line number Diff line change
@@ -1,8 +1,7 @@
segmentor_name: MTUPerNet
segmentor_name: MTUPerNet_regress
task_name: regression
binary: False
multi_temporal_strategy: linear
# time_frames: 2
time_frames: 12
#task_model_args:
#num_frames: 1
#mt_strategy: "ltae" #activated only when if num_frames > 1
Expand Down

0 comments on commit c369971

Please sign in to comment.