错误提示
AttributeError Traceback (most recent call last)
/tmp/ipykernel_2743251/865336216.py in <module>
5 train_losses = list()
6 for epoch in range(epochs):
----> 7 loss = train(train_loader,net, optimizer, epoch)
8 train_losses += loss
9 # print("train_losses=",train_losses)
/tmp/ipykernel_2743251/1587444429.py in train(dataloader, net, optimizer, epoch)
17 # 2. 输入数据给模型，进行预测
18 # print("enumerate(loader)", dataloader)
---> 19 for batch_idx, (img_deformation, img_texture, mask, mask_boundary) in enumerate(dataloader):
20 output_mask_texture, output_edge_texture = net(img_texture)
21 output_mask_deformation, output_edge_deformation = net(img_deformation)
~/anaconda3/envs/old/lib/python3.7/site-packages/jittor/dataset/dataset.py in __iter__(self)
501 if self._disable_workers:
502 self.num_workers = 0
--> 503 index_list = self._get_index_list()
504
505 if not hasattr(self, "workers") and self.num_workers:
~/anaconda3/envs/old/lib/python3.7/site-packages/jittor/dataset/dataset.py in _get_index_list(self)
436 # consist shuffle list
437 # index_list = get_random_list(self.total_len)
--> 438 index_list = self._shuffle_rng.permutation(range(self.total_len))
439
440 # scatter index_list for all mpi process
AttributeError: 'MyDataset' object has no attribute '_shuffle_rng'
前面已经报了几个类似的错误，然后我在我的数据集类上加上几个属性，如：
self._disable_workers = True
self.sampler = False
self.shuffle = self.exp_args.istrain
参考了几个新的示例也解决不了问题。
我对计图数据加载不熟悉，如果作者能够帮忙debug，我将发我的源码给您，十分感谢。