File size: 573 Bytes
3eb682b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
"""
A script to benchmark data loading.
"""
import timesformer.utils.logging as logging
from timesformer.utils.benchmark import benchmark_data_loading
from timesformer.utils.misc import launch_job
from timesformer.utils.parser import load_config, parse_args
logger = logging.get_logger(__name__)
def main():
args = parse_args()
cfg = load_config(args)
launch_job(
cfg=cfg, init_method=args.init_method, func=benchmark_data_loading
)
if __name__ == "__main__":
main()
|