Traceback (most recent call last):
File "/tmp/huangsibo/srgan/train_srgan.py", line 26, in
from dataset import CUDAPrefetcher, TrainValidImageDataset, TestImageDataset
File "/tmp/huangsibo/srgan/dataset.py", line 32, in
class TrainValidImageDataset(Dataset):
File "/tmp/huangsibo/srgan/dataset.py", line 57, in TrainValidImageDataset
def getitem(self, batch_index: int) -> [dict[str, Tensor], dict[str, Tensor]]:
TypeError: 'type' object is not subscriptable
Traceback (most recent call last): File "/tmp/huangsibo/srgan/train_srgan.py", line 26, in
from dataset import CUDAPrefetcher, TrainValidImageDataset, TestImageDataset
File "/tmp/huangsibo/srgan/dataset.py", line 32, in
class TrainValidImageDataset(Dataset):
File "/tmp/huangsibo/srgan/dataset.py", line 57, in TrainValidImageDataset
def getitem(self, batch_index: int) -> [dict[str, Tensor], dict[str, Tensor]]:
TypeError: 'type' object is not subscriptable