Xin Kong

How much GPU memory is used?

2

Hi! I set the sa_module as [4096, 1024, 256, 64] and get GPU memory exhausted error. I'm using TITAN XP 12G GPU. I want to know how much GPU memory...

BATCH_SIZE is hard-coded

1

Hi! I notice that the BATCH_SIZE is hard-coded to 1 and you will change that later. Have you fix this problem? If yes, would you update the code? If no,...

Bus error (core dumped)

29

Hi! I met an error when running the code. I have tried this both with cpu and gpu. ``` Initialated Mask RCNN network... Creating net instance... Loading net parameters... Bus...

Train ensemble models with vmap

3

Thanks a lot for developing functorch! I checked the tutorial and tried vmap to ensemble multi models. I'm curious can I use vmap to do multi-model training? For example, can...

Add some ROS nodes for visualization. A visualize demo picture is updated in README.md.

I try to trian the model with one GPU (TITAN XP with 12G memory) and get tensorflow error: Resource exhausted: OOM. Thus, I want to know how much GPU memory...

Citing errors

1

Could you check your citing format again? I get errors when cite by latex in bib format. Thanks!

Fix some small errors after running the code.

Hi, thanks for opening the source code! I'm just curious about the acceleration speed of the cuda implementation compared to pytorch.bmm operation, if the input to each MLPs are equal....

How to fix label color in point_cloud2 msg?

1

Hi! I notice that the color of the point cloud label in rviz is changed. Could we fix the color with label value? For example, label 0 is red, label...