wassname/ETSformer
1
0
Fork 0
mirror of https://github.com/wassname/ETSformer.git synced 2026-09-11 11:51:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

return attention weights

Browse Source
This commit is contained in:
gorold
2022-09-28 12:50:29 +08:00
parent b6f322dfa7
commit e7650a3e3f
4 changed files with 107 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
run.py
+1
View File
@@ -57,6 +57,7 @@ parser.add_argument('--std', type=float, default=0.2)
parser.add_argument('--smoothing_learning_rate', type=float, default=0, help='optimizer learning rate')
parser.add_argument('--damping_learning_rate', type=float, default=0, help='optimizer learning rate')
parser.add_argument('--output_attention', type=bool, default=False)
# optimization
parser.add_argument('--optim', type=str, default='adam', help='optimizer')