Dohee Kim

Is not work about 'dict' input?

3

Hello, I'm trying to use pytorch-summary But, my model's input is "dict" type. What can I do for using this library about dict input? Thank you for sharing awesome library...

CUDA version

2

Hello, Can you please let me know what version of cuda you used? The first case, I used torch==0.4.1 and cuda 10.0. When I run the command `python3 scripts/train_rpn_3d.py --config=kitti_3d_base...