-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.yaml
More file actions
14 lines (12 loc) · 935 Bytes
/
config.yaml
File metadata and controls
14 lines (12 loc) · 935 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
compress: True # set True to compress the gradient.
divide_groups: True # set True to divide the nodes into groups.
num_groups: 2 # If the divide_groups is set to True, mention the number of groups to be divided into.
start_dividing_after: 100 # Denotes the number of batches after which the asynchronous training shall start.
synchronize_weights: True # set True to synchronize the weights of the model.
synchronization_of_weights_freq: 21 # number of updates after which synchronization of weights shall be performed.
normal_synchronization: False # set True if synchronization of weights shall not be approximated.
compressor_conf: #only used when compress is set to True
rank: 1 # lower rank => more aggressive compression
min_compression_rate: 2 # skip compression on some gradients
num_iters_per_step: 1 # lower number => more aggressive compression
start_compressing_after_num_steps: 0