DongChunGuang

icon company MoZhe Security SZ icon location ShenZhen, China. icon location This guy is too lazy with nothing left.

Results 1 comments of DongChunGuang

stuck in the evaluation_loop of trainer.py when training

@sgugger I have located the problem statement: ``` trainer.py def _pad_across_processes(self, tensor, pad_index=-100): .... # Gather all sizes size = torch.tensor(tensor.shape, device=tensor.device)[None] sizes = self._nested_gather(size).cpu() .... ``` Stuck at .cpu()...