From 94f05efdbdd056ad97013d89b9323a8c2935b777 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Thu, 22 Aug 2024 15:37:43 +0200
Subject: [PATCH] TAPE encoder transformer model
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index db984b4..fcc9074 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -207,7 +207,7 @@ def main(args):
print('\nData loaded')
- model = Model_Common_Transformer(encoder_ff=args.encoder_ff, decoder_rt_ff=args.decoder_rt_ff,
+ model = Model_Common_Transformer_TAPE(encoder_ff=args.encoder_ff, decoder_rt_ff=args.decoder_rt_ff,
decoder_int_ff=args.decoder_int_ff
, n_head=args.n_head, encoder_num_layer=args.encoder_num_layer,
decoder_int_num_layer=args.decoder_int_num_layer,
--
GitLab