iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update NATS-Bench codes

Browse Source
This commit is contained in:
D-X-Y 2021-01-25 21:50:47 +08:00
parent 06d5c547c9
commit af20164814
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
exps/NATS-Bench/Analyze-time.py
View File

@ -5,11 +5,7 @@
##############################################################################
# python ./exps/NATS-Bench/Analyze-time.py #
##############################################################################
import os, sys, time, tqdm, torch, random, argparse
from typing import List, Text, Dict, Any
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
from copy import deepcopy
import os, sys, time, tqdm, argparse
from pathlib import Path
lib_dir = (Path(__file__).parent / '..' / '..' / 'lib').resolve()