Setup model and optimizer.
(args)
| 76 | |
| 77 | |
| 78 | def setup_model(args): |
| 79 | """Setup model and optimizer.""" |
| 80 | |
| 81 | model = get_model(args) |
| 82 | |
| 83 | if args.load is not None: |
| 84 | _ = load_checkpoint( |
| 85 | model, None, None, args) |
| 86 | |
| 87 | return model |
| 88 | |
| 89 | def get_masks_and_position_ids(data, |
| 90 | eod_token, |
no test coverage detected