Пример #1
0
def embed_params():
    """(inspecs, n_inputs, n_features)"""
    params = []
    params.append(([
        Inspec((64, ), I.UniformIntInitializer((0, 40000)), need_grad=False)
    ], 40000, 256))
    return params
Пример #2
0
def pairwise_inspecs_params(label_init=I.UniformIntInitializer((0, 2))):
    inspecs = []
    inspecs.append([Inspec((64, 1000)), Inspec((64, 1000), label_init, False)])
    inspecs.append([
        Inspec((64, 32, 224, 224)),
        Inspec((64, 32, 224, 224), label_init, False)
    ])
    inspecs.append([
        Inspec((64, 128, 56, 56)),
        Inspec((64, 128, 56, 56), label_init, False)
    ])
    return inspecs