How to Eliminate Warnings While Creating VGG_11 Trace File

jit
1

I try to create trace file of a model without any warning. I use the model VGG_11 at this link. I trained the model with my custom dataset then saved the model as a .pth file. Then, I loaded this .pth file and created trace file as a .pt file. Then, it gave those warnings:

VGG_Trace_Warnings
VGG_Trace_Warnings.PNG1853×313 29.7 KB

Could you offer a solution about how to eliminate those warnings?

2

The warning is telling you that there are non-deterministic nodes in your model, specifically dropouts. If you’re tracing a model that you already trained, you may have forgotten to set eval() on the module.