06 2024 档案
摘要:查看cuda版本的指令: nvidia-smi ncvv -V python -c "import torch; print(torch.version.cuda)" nvidia-smi (base) liuliu@liuliu-Legion-Y9000P-IAH7H:~/EchoFusion$
阅读全文
摘要:1 前言 记录一下Pytorch中对tensor合并分割的方法 2 合并 Pytorch中对tensor合并的方法有两种: torch.cat() torch.stack() 其中,torch.cat()直接将两个变量进行拼接,不会产生新的维度 而torch.stack()则会将tensor堆叠,产
阅读全文
摘要:1 前言 在学习mmdetection3d过程中,发现很多类后都有一个__rper__的方法,因此记录其用法 2 方法 类方法中的__repr__和__str__都是用于显示的,其中__str__面对用户,而__repr__面对开发者 class Test: def __repr__(self):
阅读全文