mirrors / dmlc--dgl

文件历史

4,090 提交 34 分支 42 Git 标签
2.2.x
提交图

3 次代码提交

作者 SHA1 备注 提交日期
Hongzhi (Steve), Chen be8763fa65 [Misc] Black auto fix. (#4679) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2022-10-08 11:59:37 +08:00
esang 4fb50be4d9 [Model] Point transformer (#3284) 
* some modifications for pointnet2

* temporarily save changes

* move files to new directory point_transformer

* implement point transformer for classification

* restore train_cls in pointnet

* implement point transformer for partseg

* fix point transformer for nan loss

* modify point transformer for cls

* modify training setting

* update transformer for cls

* update code

* update code for latest performance

* update the example index

* some minor changes

Co-authored-by: Tong He <hetong007@gmail.com>
 2021-08-27 14:33:18 +08:00
Tong He 558673e139 [Model] PointNet and PointNet++ for point cloud (#1510) 
* commit patch

* commit patch

* pointnet basic

* fix data

* reorg

* reorg

* temp status

* remove validate set

* add partseg data and model

* partseg miou

* clean up

* fix loss

* network definition match paper

* fix

* fix miou

* update data format

* fix

* fix

* working pointnet ssg cls

* avoid some pytorch bug

* fix script

* update hyperparams

* add msg module

* try different dataset

* update new dataset info

* quick fix to subgraph

* fix speed

* update training

* update

* fix bs

* update docstring

* update

* update

* remove parallel reduction in fps

* switch to kernel fps, training is 30% faster

Co-authored-by: Ubuntu <ubuntu@ip-172-31-20-181.us-west-2.compute.internal>
 2020-06-22 19:50:40 +08:00