PA3 Point Cloud Classification

基于 Jittor 的 ModelNet40 点云分类作业实现，主脚本为 pct.py。

项目结构

• pct.py：训练、预测、断点续训主脚本
• pct模板.py：保留模板结构的参考版本
• data/：数据集目录
• sitecustomize.py：默认关闭 Jittor 多进程编译
• result.json：最终测试集预测结果

环境说明

• Python 3
• Jittor
• NVIDIA CUDA 环境
• 建议 g++-12 与 CUDA 12.2 搭配使用

如果当前环境里 Jittor 的 CUDA 编译不稳定，脚本已经默认设置：

运行方式

从项目根目录运行：

python pct.py

常用参数：

python pct.py --epochs 200 --batch_size 16 --lr 0.001

断点续训

脚本会自动尝试读取 pct_checkpoint.pkl，并从上次完成的 epoch 继续训练。

重新开始训练时，如果不想加载旧状态，删除以下文件：

pct_checkpoint.pkl
pct_model*.pkl
result_*.json

输出文件

每 10 个 epoch 会保存一次：

• pct_checkpoint.pkl
• pct_model_epoch{N}.pkl
• result_epoch{N}.json

训练结束后还会额外保存：

• pct_model.pkl
• result.json

提交说明

提交时通常只需要：

• result.json

如果平台要求压缩包，可手动打包为：

result.zip
└── result.json