sczhou / ProPainter

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting
https://shangchenzhou.com/projects/ProPainter/
Other
5.45k stars 645 forks source link

Training phase #87

Open 1chuanchuan opened 1 month ago

1chuanchuan commented 1 month ago

`(propainter) PS E:\gzh\Project\watermark-remove> python train.py -c configs/train_propainter.json world_size: 1 [**] create folder experiments_model/propainter_train_propainter Traceback (most recent call last): File "train.py", line 105, in mp.spawn(main_worker, nprocs=torch.cuda.device_count(), args=(config, )) File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 239, in spawn return start_processes(fn, args, nprocs, join, daemon, start_method='spawn') File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 197, in start_processes while not context.join(): File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 160, in join raise ProcessRaisedException(msg, error_index, failed_process.pid) torch.multiprocessing.spawn.ProcessRaisedException:

-- Process 0 terminated with the following error: Traceback (most recent call last): File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 69, in _wrap fn(i, *args) File "E:\gzh\Project\watermark-remove\train.py", line 74, in main_worker trainer = core.dict[trainer_version].dict'Trainer' self.train_dataset = TrainDataset(config['train_data_loader']) File "E:\gzh\Project\watermark-remove\core\dataset.py", line 43, in init frame_list = sorted(os.listdir(os.path.join(self.video_root, v))) FileNotFoundError: [WinError 3] 系统找不到指定的路径。: 'E:\gzh\Project\watermark-remove\datasets\003234408d'

(propainter) PS E:\gzh\Project\watermark-remove> python train.py -c configs/train_propainter.json world_size: 1 [] create folder experiments_model/propainter_train_propainter Pretrained flow completion model has loaded... Network [InpaintGenerator] was created. Total number of parameters: 39.4 million. To see the architecture, do print(network). Warnning: There is no trained model found.An initialized model will be used. 0%| | 0/700000 [00:00<?, ?it/s]D :\Anaconda3\envs\propainter\lib\site-packages\torch\functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\TensorShape.cpp:3484.) return _VF.meshgrid(tensors, kwargs) # type: ignore[attr-defined] 0%| | 0/700000 [01:15<?, ?it/s] Traceback (most recent call last): File "train.py", line 105, in mp.spawn(main_worker, nprocs=torch.cuda.device_count(), args=(config, )) File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 239, in spawn return start_processes(fn, args, nprocs, join, daemon, start_method='spawn') File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 197, in start_processes while not context.join(): File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 160, in join raise ProcessRaisedException(msg, error_index, failed_process.pid) torch.multiprocessing.spawn.ProcessRaisedException:

-- Process 0 terminated with the following error: Traceback (most recent call last): File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\multiprocessing\spawn.py", line 69, in _wrap fn(i, *args) File "E:\gzh\Project\watermark-remove\train.py", line 76, in main_worker trainer.train() File "E:\gzh\Project\watermark-remove\core\trainer.py", line 345, in train self._train_epoch(pbar) File "E:\gzh\Project\watermark-remove\core\trainer.py", line 377, in _train_epoch prop_imgs, updated_local_masks = self.netG.module.img_propagation(masked_local_frames, pred_flows_bi, local_masks, interpolation=self.interp_mode) File "D:\Anaconda3\envs\propainter\lib\site-packages\torch\nn\modules\module.py", line 1614, in getattr raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'InpaintGenerator' object has no attribute 'module'` Hello, I encountered this problem when running the code, could you please tell me which file is the training model and where it is located?