(args)
| 689 | |
| 690 | |
| 691 | def build_model(args): |
| 692 | return DeciWatchTransformer( |
| 693 | input_nc=args['input_dim'], |
| 694 | decoder_hidden_dim=args['decoder_hidden_dim'], |
| 695 | encoder_hidden_dim=args['encoder_hidden_dim'], |
| 696 | dropout=args['dropout'], |
| 697 | nhead=args['nheads'], |
| 698 | dim_feedforward=args['dim_feedforward'], |
| 699 | num_encoder_layers=args['enc_layers'], |
| 700 | num_decoder_layers=args['dec_layers'], |
| 701 | activation=args['activation'], |
| 702 | pre_norm=args['pre_norm'], |
| 703 | ) |
| 704 | |
| 705 | |
| 706 | def _get_activation_fn(activation): |
no test coverage detected