Skip to content

Commit

Permalink
readme
Browse files Browse the repository at this point in the history
  • Loading branch information
lucidrains authored Sep 11, 2023
1 parent 977ee55 commit b7ace1b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ out, total_aux_loss, balance_loss, router_z_loss = moe_block(inputs) # (4, 1024,

# the total auxiliary loss will need to be summed and then added to the main loss

# the other two losses are the breakdown, weighed by the coefficients
# the other two losses are the unweighted breakdown for logging purposes
```

## Todo
Expand Down

0 comments on commit b7ace1b

Please sign in to comment.