Commit e21b85ff authored by mathpluscode's avatar mathpluscode

update config

parent 6069e5a7
Pipeline #3263 passed with stage
in 1 minute and 11 seconds
......@@ -11,7 +11,7 @@ data:
shape: # shape is [height, width]
orig: [540, 1920]
border: 130 # crop border manually, output has shape [540, 1660]
input: [64, 64] # input size for NN
input: [128, 384] # input size for NN
sl:
keep_ratio: 1 # proportion of labeled data, randomly sampled
ssl:
......@@ -51,7 +51,7 @@ model:
lr: 1.e-4 # learning rate
clip_norm: 5.0 # grad clip norm
weight_decay: 1.e-5 # l2 loss weight for adam or weight_decay for adamw
batch_size: 4
batch_size: 64
ssl:
mode: "mean_teacher" # pi: consistency between noised1 and noised2, mean_teacher: noise1 for student and noise2 for teacher
loss:
......@@ -66,7 +66,7 @@ model:
tf: # configuration for tensorflow
gpu: "" # GPU ID should be specified manually
num_parallel_calls: 4 # number of cpus
num_parallel_calls: 8 # number of cpus
save_best:
metric_name: "metric_bin/f1_percentile_50"
large_is_better: true
......@@ -77,7 +77,7 @@ tf: # configuration for tensorflow
log_step_count_steps: 100 # output frequency
train: # tf.estimator.TrainSpec
max_steps: 100
max_steps: 10000
eval: # tf.estimator.EvalSpec
steps: 200 # number of batches for evaluation, each folder has at maximum 260 frames
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment