dilated_tooth_seg_net icon indicating copy to clipboard operation
dilated_tooth_seg_net copied to clipboard

NameError: name 'fps' is not defined

Open littlezhang231 opened this issue 9 months ago • 0 comments

Hello I met this question in running this code he value of the num_workers argumenttonum_workers=11in theDataLoader` to improve performance. Sanity Checking DataLoader 0: 0%| | 0/2 [00:00<?, ?it/s]Traceback (most recent call last): File "/mnt/dilated_tooth_seg_net-main/train_network.py", line 102, in trainer.fit(model=model, train_dataloaders=train_dataloader, val_dataloaders=val_dataloader, ckpt_path=args.ckpt) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 545, in fit call._call_and_handle_interrupt( File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/call.py", line 44, in _call_and_handle_interrupt return trainer_fn(*args, **kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 581, in _fit_impl self._run(model, ckpt_path=ckpt_path) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 990, in _run results = self._run_stage() File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 1034, in _run_stage self._run_sanity_check() File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 1063, in _run_sanity_check val_loop.run() File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/utilities.py", line 181, in _decorator return loop_run(self, *args, **kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/evaluation_loop.py", line 134, in run self._evaluation_step(batch, batch_idx, dataloader_idx, dataloader_iter) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/evaluation_loop.py", line 391, in _evaluation_step output = call._call_strategy_hook(trainer, hook_name, *step_args) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/call.py", line 309, in _call_strategy_hook output = fn(*args, **kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/strategies/strategy.py", line 403, in validation_step return self.lightning_module.validation_step(*args, **kwargs) File "/mnt/dilated_tooth_seg_net-main/models/dilated_tooth_seg_network.py", line 109, in validation_step pred = self.model(x, pos) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, **kwargs) File "/mnt/dilated_tooth_seg_net-main/models/dilated_tooth_seg_network.py", line 63, in forward x1, _ = self.dilated_edge_graph_conv_block1(x, pos, cd=cd) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, **kwargs) File "/mnt/dilated_tooth_seg_net-main/models/layer.py", line 229, in forward idx_fps = fps(pos.reshape(B * N, -1)[idx_l], self.k).long() NameError: name 'fps' is not defined

How can I solve it It happpens after processing data

littlezhang231 avatar May 05 '24 11:05 littlezhang231