Skip to content

Commit

Permalink
Removed arithmetic op
Browse files Browse the repository at this point in the history
Signed-off-by: Selvaraj Anandaraj <[email protected]>
  • Loading branch information
Selvaraj Anandaraj committed Jan 23, 2024
1 parent 8ddfeeb commit 3065ddf
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ model:

##Offloading Activations/Weights to CPU
cpu_offloading: False
cpu_offloading_num_layers: ${subtract:${num_layers},1} #This value should be between [1,num_layers-1] as we don't want to offload the final layer's activations and expose any offloading duration for the final layer
cpu_offloading_num_layers: 11 #This value should be between [1,num_layers-1] as we don't want to offload the final layer's activations and expose any offloading duration for the final layer
cpu_offloading_activations: True
cpu_offloading_weights: True

Expand Down

0 comments on commit 3065ddf

Please sign in to comment.