(number, threshold=1e-2)
| 93 | |
| 94 | |
| 95 | def truncate_number(number, threshold=1e-2): |
| 96 | # avoid overflow with AMP training |
| 97 | return TruncateFunction.apply(number, threshold) |
| 98 | |
| 99 | |
| 100 | def get_named_linears(module, type): |
nothing calls this directly
no outgoing calls
no test coverage detected