huzaifanafees's picture
Upload 4 files
97fe9c2
raw
history blame
293 Bytes
class Config:
vocab_size = 28000 # Vocabulary Size
sequence_length = 35
batch_size = 128
validation_split = 0.20
embed_dim = 300
num_layers = 4
latent_dim = 2048
num_heads = 12
epochs = 50 # Number of Epochs to train
is_training = True
config = Config()