cooperative-foundational-models

cooperative-foundational-models copied to clipboard

how to train

3

hello, i am wodering how can i train the model on my own datasets

KDgggg

running inference on single image overloads gpu memory

3

Hi, When trying to run inference on a test image using your script I get a "CUDA out of memory error". My image size is 640x480, and I have a...

shahyaan

(inference_single_image) RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

4

Thank you so much for updating "inference_single_image". But, the following error occurred when using the script. `[02/21 14:08:53 fvcore.common.checkpoint]: [Checkpointer] Loading from weights/maskrcnn_v2/model_final.pth ... Downloading (…)ip_pytorch_model.bin: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3.51G/3.51G [04:00

ralpyna

Add inference code for user-defined data

4

Thanks for your great work. Can you add a script for model inference on a single input image? Thanks in advance

chau25102001

Has this paper been published?

kunpeng337