未验证 提交 3a3ff75c 编写于 作者: L LielinJiang 提交者: GitHub

Fix unittest random failed of test_datasets (#30804)

* fix test_datasets unittest
上级 04532b8a
...@@ -128,6 +128,13 @@ class Flowers(Dataset): ...@@ -128,6 +128,13 @@ class Flowers(Dataset):
scio = try_import('scipy.io') scio = try_import('scipy.io')
# double check data download
self.label_file = _check_exists_and_download(self.label_file, LABEL_URL,
LABEL_MD5, 'flowers', True)
self.setid_file = _check_exists_and_download(self.setid_file, SETID_URL,
SETID_MD5, 'flowers', True)
self.labels = scio.loadmat(self.label_file)['labels'][0] self.labels = scio.loadmat(self.label_file)['labels'][0]
self.indexes = scio.loadmat(self.setid_file)[self.flag][0] self.indexes = scio.loadmat(self.setid_file)[self.flag][0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册