Innan666
Innan666
I got an error is torch.cuda.is_available() is False, but I have GPU. Could you figure out, what is the reason? total parameters: 116969339 loading from c:\Users\IFL\Desktop\Liu\Detector\CornerNet-Lite-master\core\..\cache\nnet\CornerNet_Saccade\CornerNet_Saccade_500000.pkl Traceback (most recent call...
(measure) C:\Users\IFL\Desktop\Liu\Detector\CenterNet\src>python demo.py ctdet --demo webcam --load_model ../models/ctdet_coco_dla_2x.pth Fix size testing. training chunk_sizes: [32] The output will be saved to C:\Users\IFL\Desktop\Liu\Detector\CenterNet\src\lib\..\..\exp\ctdet\default heads {'hm': 80, 'wh': 2, 'reg': 2} Creating model......
(measure) C:\Users\IFL\Desktop\Liu\Detector\CenterNet\src>python main.py ctdet --exp_id 640 --batch_size 4 --dataset corner --input_res 640 --num_epochs 150 --val_intervals 10 Fix size testing. training chunk_sizes: [4] The output will be saved to C:\Users\IFL\Desktop\Liu\Detector\CenterNet\src\lib\..\..\exp\ctdet\640 heads...
这个里面有手动设置learningrate的地方吗