def debug_sparse_mlp_model(args, model, input_data):
hiddens = [int(s) for s in args.hidden.split('-')] + [2]
sums = debug_sparse_mlp(model, input_data.data, args.input_dim, hiddens,
input_data.sparse_segments)
return BatchLRLoss(model, sums, input_data.label)
def mlp_decomp_model(args, model, input_data):
hiddens = [int(s) for s in args.hidden.split('-')] + [2]
sums = mlp_decomp(model, input_data.data, args.input_dim, hiddens)
return BatchLRLoss(model, sums, input_data.label)
