util.py 691 字节
Newer Older
1
"""
Y
Yibing Liu 已提交
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
    """
23 24 25
    if not os.path.isdir(path): 
        if os.path.split(path)[0]:
            mkdir(os.path.split(path)[0])
Y
Yibing Liu 已提交
26 27 28 29 30
    else:
        return
    os.mkdir(path)


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


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