#516 update torchaudio and fix dataloader init

Merged
Erpim merged 23 commits from torchaudio into master 1 year ago
hanjr commented 1 year ago
Erpim reviewed 1 year ago
@@ -331,2 +331,3 @@
raise TypeError(default_collate_err_msg_format.format(elem_type))
raise TypeError(default_collate_err_msg_format.format(elem_type))

Erpim commented 1 year ago
直接做累加?不用返回list再加了吧?
Erpim reviewed 1 year ago
@@ -333,0 +338,4 @@
for i in data:
collate_nbytes(i, return_list)
elif isinstance(data, collections.abc.Mapping):
for key in data:
Erpim commented 1 year ago
collate_nbytes(data[key], return_list)
hanjr marked this conversation as resolved
Erpim reviewed 1 year ago
msadapter/pytorch/utils/data/dataloader.py
@@ -380,0 +395,4 @@
collate_nbytes(single_data, nbytes)
del temp_dataloaderiter
del single_data
nbytes = int((sum(nbytes) >> 20) * 1.2)
Erpim commented 1 year ago
魔鬼数字,注释下为什么是1.2,建议新建个变量,可以叫relaxation ratio之类的。
Erpim reviewed 1 year ago
msadapter/pytorch/utils/data/dataloader.py
@@ -380,0 +399,4 @@
self.maxsize = MIN_POWER_TWO(nbytes)
else:
self.maxsize = 128

Erpim commented 1 year ago
初始化的时候就创建迭代器对象
Erpim reviewed 1 year ago
@@ -8,3 +1,1 @@
# from .common import AudioMetaData
#
#

Erpim commented 1 year ago
多余空行
hanjr marked this conversation as resolved
Erpim merged commit 2416936c7b into master 1 year ago
The pull request has been merged as 2416936c7b.
Sign in to join this conversation.
No reviewers
No Label
No Milestone
No Assignees
2 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Loading…
There is no content yet.