(model)
| 59 | return data_loader |
| 60 | |
| 61 | def init_weights(model): |
| 62 | for name, param in model.named_parameters(): |
| 63 | if "embedding" not in name: |
| 64 | torch.nn.init.uniform_( |
| 65 | param, a=-WEIGHT_INIT_RANGE, b=WEIGHT_INIT_RANGE |
| 66 | ) |
| 67 |
nothing calls this directly
no outgoing calls
no test coverage detected