run system with CPU only

[daynial132]

trafficstars

hello,

I dont have GPU right now but wants to try your example, is there any way to do so kind regard's

[MyraBaba]

@daynial132

change device to :

device = torch.device("cuda:0" if torch.cuda.is_available()else "cpu")

[linearpi]

change device and change laod """lprnet.load_state_dict(torch.load(args.pretrained_model))"""

change load to :

lprnet.load_state_dict(torch.load(args.pretrained_model, map_location=torch.device('cpu')))