"torchvision/transforms/_presets.py" did not exist on "2b5ab1bc5884c621a19e48e4f0ea3d82c4907807"
bm_mesh_normal_consistency.py 980 Bytes
Newer Older
Patrick Labatut's avatar
Patrick Labatut committed
1
2
3
4
5
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
facebook-github-bot's avatar
facebook-github-bot committed
6
7
8


from itertools import product
9

facebook-github-bot's avatar
facebook-github-bot committed
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import torch
from fvcore.common.benchmark import benchmark
from test_mesh_normal_consistency import TestMeshNormalConsistency


def bm_mesh_normal_consistency() -> None:
    devices = ["cpu"]
    if torch.cuda.is_available():
        devices.append("cuda")

    kwargs_list = []
    num_meshes = [16, 32, 64]
    levels = [2, 3]
    test_cases = product(num_meshes, levels, devices)
    for case in test_cases:
        n, l, d = case
        kwargs_list.append({"num_meshes": n, "level": l, "device": d})

    benchmark(
        TestMeshNormalConsistency.mesh_normal_consistency_with_ico,
        "MESH_NORMAL_CONSISTENCY_ICO",
        kwargs_list,
        warmup_iters=1,
    )
Christoph Lassner's avatar
Christoph Lassner committed
34
35
36
37


if __name__ == "__main__":
    bm_mesh_normal_consistency()