环境安装

pip install jittor

数据准备

数据应放在一个独立的文件夹中，包含预处理好的数据，train_points.npy, train_labels.npy, test_points.npy，脚本中可以用--data_dir参数指定数据文件夹路径。

训练

可用-o参数指定优化器类型，可选sgd, adam, adamw

python pct.py -o adamw

评测

用-o参数指定输出路径，--ckpt参数指定模型参数路径

python eval.py --ckpt ./ckpt/ckpt.pkl -o result.json