diff --git a/config.py b/config.py new file mode 100644 index 0000000..32b57d1 --- /dev/null +++ b/config.py @@ -0,0 +1,21 @@ +import json + +import torch + +with open('vocab/vocabulary.json', 'r', encoding='utf-8') as f: + vocab = json.load(f) + vocab_size = len(vocab) +batch_size=1 +input_dim=1024 +embedding_dim=input_dim +dropout=0.1 +n_head=32 +Encoder_n_layers=1 +d_ff1=input_dim*4 +d_ff2=input_dim*4 +Decoder_n_layer=0 +hidden_dim=input_dim +seq_len=30 +bias=True +device=torch.device('cuda') +print(torch.cuda.is_available()) \ No newline at end of file