提交 a5b21912 编写于 作者: L LDOUBLEV

fix conflicts

上级 7cce85cc
......@@ -57,9 +57,9 @@ class SimpleDataSet(Dataset):
sample_num = round(len(datas) * sample_ratio)
if data_num_per_epoch is not None:
sample_num = data_num_per_epoch * sample_ratio
sample_num = int(data_num_per_epoch * sample_ratio)
nums, rem = sample_num // len(datas), sample_num % len(datas)
nums, rem = int(sample_num // len(datas)), int(sample_num % len(datas))
return list(datas) * nums + random.sample(datas, rem)
def get_image_info_list(self,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册