how to directly run stage2 mvcnn

[zhangdongwei1998]

Thanks for your great work, I wonder how to directly run stage2 mvcnn without run stage1 first. I would appreciate if you could give some advice.

[YY-MyWorld]

Traceback (most recent call last):
File "train_mvcnn.py", line 79, in
trainer.train(1)
File "/home/wanghuijuan1/MVCNN/mvcnn_pytorch-master/tools/Trainer.py", line 50, in train
for i, data in enumerate(self.train_loader):
File "/home/wanghuijuan1/anaconda3/envs/torch18/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 517, in next
data = self._next_data()
File "/home/wanghuijuan1/anaconda3/envs/torch18/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 557, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/home/wanghuijuan1/anaconda3/envs/torch18/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/wanghuijuan1/anaconda3/envs/torch18/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/wanghuijuan1/MVCNN/mvcnn_pytorch-master/tools/ImgDataset.py", line 76, in getitem
class_id = self.classnames.index(class_name)
ValueError: 'train' is not in list
stage2 mvcnn unable to execute，This problem occurred, did you have this problem? How did you solve it, thanks a lot