mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Reorg benchmark.py import

Browse Source
This commit is contained in:
Ross Wightman 2022-07-07 15:21:29 -07:00
parent 500c190860
commit 4670d375c6
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
benchmark.py
View File

@ -6,24 +6,23 @@ An inference and train step benchmark script for timm models.
Hacked together by Ross Wightman (https://github.com/rwightman) Hacked together by Ross Wightman (https://github.com/rwightman)
""" """
import argparse import argparse
import os
import csv import csv
import json import json
import time
import logging import logging
import torch import time
import torch.nn as nn
import torch.nn.parallel
from collections import OrderedDict from collections import OrderedDict
from contextlib import suppress from contextlib import suppress
from functools import partial from functools import partial
import torch
import torch.nn as nn
import torch.nn.parallel
from timm.data import resolve_data_config
from timm.models import create_model, is_model, list_models from timm.models import create_model, is_model, list_models
from timm.optim import create_optimizer_v2 from timm.optim import create_optimizer_v2
from timm.data import resolve_data_config
from timm.utils import setup_default_logging, set_jit_fuser from timm.utils import setup_default_logging, set_jit_fuser
has_apex = False has_apex = False
try: try:
from apex import amp from apex import amp