Runtime error when trying demo

[AissaGeek]

Hello, I'm trying to excute demo.py on google collab, it always gets a runtime error, i may solve this by reducing the batch size, but can you show me how ? Thank you

[vlad3996]

Use with torch.no_grad(): to inference. You can also try my simple demo.py with original author's checkpoint https://github.com/vlad3996/FaceDetection-DSFD

[nyck33]

@vlad3996 Can you please clarify where to put torch.no_grad in demo.py? I don't see it in your code nor do I see it in this repo's demo.py

[nyck33]

in the repo's demo.py, I put it in the first line of def infer() but it didn't help.

[Liuqiyuan670]

has you solved your problem? i meet the runtimeError too