我运行了front_main.py,报错
Traceback (most recent call last):
File "e:/OCTASNet/code/OCTANet/front_main.py", line 64, in
first_net = train_first_stage(viz, writer, train_dataloader, first_net, first_optim, args.init_lr, thin_criterion, thick_criterion, device, args.power, epoch, args.first_epochs)
File "e:\OCTASNet\code\OCTANet\train.py", line 15, in train_first_stage
thin_gt = sample[2].to(device)
IndexError: list index out of range
我运行了front_main.py,报错 Traceback (most recent call last): File "e:/OCTASNet/code/OCTANet/front_main.py", line 64, in
first_net = train_first_stage(viz, writer, train_dataloader, first_net, first_optim, args.init_lr, thin_criterion, thick_criterion, device, args.power, epoch, args.first_epochs)
File "e:\OCTASNet\code\OCTANet\train.py", line 15, in train_first_stage
thin_gt = sample[2].to(device)
IndexError: list index out of range
为什么sample[2].to(device)就超范围了?