↓ 1 callersFunctionknn_point Input: nsample: max sample number in local region xyz: all points, [B, N, C] new_xyz: query points, [B, S, C] Return:
models/point_encoder.py:17
↓ 1 callersFunctiontrain(train_loader, clip_model, model, criterion, optimizer, scaler, scheduler, epoch, args)
main.py:377