Skip to content

Commit

Permalink
Note about performance
Browse files Browse the repository at this point in the history
  • Loading branch information
n-shevko committed Mar 6, 2025
1 parent fd0ef9b commit ec63566
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions docs/source/guide/guide_part_i.rst
Original file line number Diff line number Diff line change
Expand Up @@ -193,6 +193,9 @@ Example 2: Batch dimension specified via batch_size parameter (no batch axis in
weights = Weight(name='weight_feature', value=torch.rand(100, 1000), sparse=True, batch_size=2)
bias = Bias(name='bias_feature', value=torch.rand(100, 1000), sparse=True, batch_size=2)
Note that subtraction and addition operations for sparse tensors are 90 times slower than those for dense tensors (tested on RTX 3060)

Specifying monitors
*******************

Expand Down

0 comments on commit ec63566

Please sign in to comment.