news 2026/4/25 18:31:18

服务器多gpu训练CPGNet过程记录

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
服务器多gpu训练CPGNet过程记录

查看显存占用命令:
nvidia-smi --query-compute-apps=pid,process_name,used_memory --format=csv
分布式训练,指定gpu训练命令
CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --master_port=12098 --nproc_per_node=4 train.py --config config/config_cpgnet_sgd_bili_sample_ohem_fp16.py

验证:CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --master_port=12098 --nproc_per_node=4 evaluate.py --config config/config_cpgnet_sgd_bili_sample_ohem_fp16.py --start_epoch 0 --end_epoch 47

配置文件中的范围修改为实际的大小:
class Voxel:
RV_theta = (-40.0, 20.0)
range_x = (-50.0, 50.0)
range_y = (-50.0, 50.0)
range_z = (-3.0, 5.0)

bev_shape = (600, 600, 30) rv_shape = (64, 2048)

修改为:
class Voxel:
RV_theta = (-10.0, 60.0)
range_x = (-20.0, 50.0)
range_y = (-20.0, 20.0)
range_z = (-2.0, 3.0)

bev_shape = (600, 600, 30) rv_shape = (64, 2048)
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/25 18:29:47

Voxtral-4B-TTS-2603开源可部署:Mistral官方权重+社区Web封装完整溯源

Voxtral-4B-TTS-2603开源可部署:Mistral官方权重社区Web封装完整溯源 1. 平台介绍 Voxtral-4B-TTS-2603是Mistral发布的开源权重语音合成(TTS)模型,专为语音Agent等生产场景设计。这个模型支持多语言文本转语音功能,并提供多种预设音色选择…

作者头像 李华
网站建设 2026/4/25 18:25:20

Phi-4-mini-reasoning部署案例:边缘服务器(Jetson AGX Orin)可行性评估

Phi-4-mini-reasoning部署案例:边缘服务器(Jetson AGX Orin)可行性评估 1. 项目背景与模型概述 Phi-4-mini-reasoning是微软推出的3.8B参数轻量级开源模型,专为数学推理、逻辑推导和多步解题等强逻辑任务设计。这款模型主打&quo…

作者头像 李华
网站建设 2026/4/25 18:22:22

前端小白也能搞定!管理大屏实战:视频监控、BIM、一次图全解

本文致力于用最接地气的方式,帮你快速上手BIM、3D可视化与实时监控项目。导读 你好!如果你是一名刚接触监控直播、楼宇等工业可视化领域的前端开发,面对“BIM模型”、“实时监控”、“一次图”这些陌生词汇感到手足无措,那么这篇文…

作者头像 李华
网站建设 2026/4/25 18:20:36

深度学习学习率配置与优化策略详解

1. 学习率在深度学习中的核心作用第一次训练神经网络时,我盯着损失曲线那个毫无波动的直线整整两小时,才意识到问题出在学习率上——它被设成了0.000001。这个参数看似简单,实则是深度学习训练中最需要精细调控的"温度计"。它决定了…

作者头像 李华