File size: 1,205 Bytes
d4b77ac
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
### General settings
name: LAFC_train
use_tb_logger: True
outputdir: /myData/ret/experiments/
datadir: /myData/
record_iter: 16

### Calling definition
model: lafc
datasetName_train: train_dataset_edge
network: network

### Datasets
datasets:
  train:
    name: youtubevos
    type: video
    mode: train
    dataInfo_config: ./config/data_info.yaml
    use_shuffle: True
    n_workers: 0
    batch_size: 4

  val:
    name: youtubevos
    type: video
    mode: val
    use_shuffle: False
    n_workers: 1
    batch_size: 1
    val_config: ./config/valid_config.yaml

### train settings
train:
  lr: 0.0001
  lr_decay: 0.1
  manual_seed: 10
  BETA1: 0.9
  BETA2: 0.999
  MAX_ITERS: 280000
  UPDATE_INTERVAL: 120000
  WARMUP: ~
  val_freq: 1  # epoch为~M~U~M, set to 1 is for debug

### logger
logger:
  PRINT_FREQ: 16
  SAVE_CHECKPOINT_FREQ: 1   # 100 is for debug consideration

### Data related parameters
num_flows: 3
flow_interval: 3
sample: seq
use_edges: 0

### Model related parameters
PASSMASK: 1
use_bias: 1
init_weights: 1
cnum: 48
resBlocks: 1
edge_residuals: 4
conv_type: vanilla
in_channel: 3
use_residual: 1
gc: 0
rescale: 1

### Loss weights
L1M: 1
sm: 1
sm2: 1
edge_loss: 1
ternary: 0.01