Bug : eval_ssd.py

[leaf918]

trafficstars

in line 151,in file eval_ssd.py,the var device assigned to CPU,the inference will NOT work, to fix it ,it should be stronger in line 37 in file ssd.py. tks.

  timer.start("Load Model")
    net.load(args.trained_model)
    net = net.to(DEVICE)
    print(f'It took {timer.end("Load Model")} seconds to load the model.')