def embed_params():
"""(inspecs, n_inputs, n_features)"""
params = []
params.append(([
Inspec((64, ), I.UniformIntInitializer((0, 40000)), need_grad=False)
], 40000, 256))
return params
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