コード例 #1
0
ファイル: CollectedTNN.py プロジェクト: ranshiju/BayesianTN
def Paras_VL_CNN1TN1():
    para = def_parameters()
    para['which_TN_set'] = 'ctnn'
    para['TN'] = 'VL_CNN1TN1'
    para['batch_size'] = 1000
    para['d'] = 2
    para['feature_map'] = '1Relu'
    para['mps_init'] = 'No.1'
    para['Lagrangian_way'] = 0
    para['Lagrangian'] = None
    para['check_time'] = 5
    para['save_time'] = 20

    para['it_time'] = 400
    para['lr'] = 1e-3
    return para
コード例 #2
0
ファイル: CollectedTNN.py プロジェクト: ranshiju/BayesianTN
def Paras_VL_NN():
    para = def_parameters()
    para['which_TN_set'] = 'cnn'
    para['TN'] = 'VL_NN'
    para['batch_size'] = 1000
    para['d'] = 2
    para['feature_map'] = 'RelSig'
    para['mps_init'] = 'rand'
    para['Lagrangian_way'] = 0
    para['Lagrangian'] = None
    para['check_time'] = 5
    para['save_time'] = 20

    para['it_time'] = 400
    para['lr'] = 1e-3
    return para
コード例 #3
0
ファイル: CollectedTNN.py プロジェクト: ranshiju/BayesianTN
def Paras_VL_TIPoolingC400_3():
    para = def_parameters()
    para['which_TN_set'] = 'ctnn'
    para['TN'] = 'VL_TIPoolingC400_3'
    para['batch_size'] = 2000
    para['d'] = 4
    para['feature_map'] = '1RelSigCos'
    para['mps_init'] = 'No.1'
    para['Lagrangian_way'] = 0
    para['Lagrangian'] = None
    para['check_time'] = 5
    para['save_time'] = 20

    para['it_time'] = 400
    para['lr'] = 1e-3
    return para
コード例 #4
0
ファイル: CollectedTNN.py プロジェクト: ranshiju/BayesianTN
def Paras_VL_Full_TI_Pooling_LargeC():
    para = def_parameters()
    para['which_TN_set'] = 'ctnn'
    para['TN'] = 'VL_Full_TI_Pooling_LargeC'
    para['batch_size'] = 2000
    para['d'] = 2
    para['feature_map'] = '1Relu'
    para['mps_init'] = 'No.1'
    para['Lagrangian_way'] = 0
    para['Lagrangian'] = None
    para['check_time'] = 5
    para['save_time'] = 20
    # para['optimizer'] = 'SGD'

    para['it_time'] = 400
    para['lr'] = 1e-3
    return para