aRI0U / RandLA-Net-pytorch

PyTorch implementation of RandLA-Net
344 stars 52 forks source link

Update train.py #10

Closed shnhrtkyk closed 4 years ago

shnhrtkyk commented 4 years ago

The batch size definition is changed "args" to "cfg".