Browse Source

CPU works为1?

master
liupengfei 1 month ago
parent
commit
4273a74965
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      default_config.yaml

+ 1
- 1
default_config.yaml View File

@@ -13,7 +13,7 @@ model: "FlowNet2" # Name of model to be loaded
rgb_max: 255 # rgb channel used
batchNorm: False # boolean switch to whether add batchnorm before conv
lr: 0.000001 # Learning rate
num_parallel_workers: 2 # Number of CPU worker used to load data
num_parallel_workers: 1 # Number of CPU worker used to load data
max_rowsize: 2 # Number of max rowsize used to load data
batch_size: 2 # Numbers of image pairs in a mini-batch
epoch_size: 1 # Total number of epochs


Loading…
Cancel
Save