sahibnanda's picture
commit
d9579cb
raw
history blame
546 Bytes
{"module": "keras_nlp.src.models.bart.bart_backbone", "class_name": "BartBackbone", "config": {"name": "bart_backbone", "trainable": true, "vocabulary_size": 50265, "num_layers": 6, "num_heads": 12, "hidden_dim": 768, "intermediate_dim": 3072, "dropout": 0.1, "max_sequence_length": 1024}, "registered_name": "keras_nlp>BartBackbone", "build_config": {"input_shape": {"encoder_token_ids": [null, null], "encoder_padding_mask": [null, null], "decoder_token_ids": [null, null], "decoder_padding_mask": [null, null]}}, "weights": "model.weights.h5"}