util.py 652 字节
Newer Older
Y
Yibing Liu 已提交
1 2 3 4
""""
Utils
"""

Y
Yibing Liu 已提交
5
import six
Y
Yibing Liu 已提交
6
import os
Y
Yibing Liu 已提交
7 8


Y
Yibing Liu 已提交
9
def print_arguments(args):
Y
Yibing Liu 已提交
10 11 12
    """
    Print arguments
    """
Y
Yibing Liu 已提交
13
    print('-----------  Configuration Arguments -----------')
Y
Yibing Liu 已提交
14
    for arg, value in sorted(six.iteritems(vars(args))):
Y
Yibing Liu 已提交
15 16 17 18
        print('%s: %s' % (arg, value))
    print('------------------------------------------------')


Y
Yibing Liu 已提交
19
def mkdir(path):
Y
Yibing Liu 已提交
20 21 22
    """
    Mkdir
    """
Y
Yibing Liu 已提交
23 24 25 26 27 28 29
    if not os.path.isdir(path):
        mkdir(os.path.split(path)[0])
    else:
        return
    os.mkdir(path)


Y
Yibing Liu 已提交
30
def pos_encoding_init():
Y
Yibing Liu 已提交
31 32 33
    """
    Pos encoding init
    """
Y
Yibing Liu 已提交
34 35 36 37
    pass


def scaled_dot_product_attention():
Y
Yibing Liu 已提交
38 39 40
    """
    Scaleed dot product attention
    """
Y
Yibing Liu 已提交
41
    pass