Update NATS-Bench to cope with lib->xautodl

2021-05-21 13:55:38 +08:00
parent b4e8eae63a
commit 97717d826e
11 changed files with 75 additions and 80 deletions
--- a/exps/NATS-Bench/show-dataset.py
+++ b/exps/NATS-Bench/show-dataset.py
@@ -8,6 +8,7 @@
 import os, sys, time, torch, random, argparse
 from typing import List, Text, Dict, Any
 from PIL import ImageFile
+
 ImageFile.LOAD_TRUNCATED_IMAGES = True
 from copy import deepcopy