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)