search

Linzaer / Ultra-Light-Fast-Generic-Face-Detector-1MB

💎1MB lightweight face detection model (1MB轻量级人脸检测模型)
MIT License
7.17k stars 1.54k forks source link

Error in resuming training #199

Closed PhenomenalOnee closed 4 years ago

PhenomenalOnee commented 4 years ago

Hello, I was training the slim model. When i try to resume the training, it causes the following error: AttributeError: 'DataParallel' object has no attribute 'load' Can you please help?

PhenomenalOnee commented 4 years ago

Got the solution, change line where its loading resume model if args.resume: logging.info(f"Resume from the model {args.resume}")

net.load(args.resume) #old line

    net.module.load(args.resume) # modification