MobileNetV3Lite failed with a RuntimeError during the first backprop stage. The error was caused in the final sequential stage of the network by a version mismatch between the ReLU and Dropout layers. Setting inplace=False in the Dropout final Dropout layer realigns the versions and enables backprop to function correctly.
The remaining changesets are required to adhere to the pre-commit formatting rules. The activation_layer argument in MobileNetV3 was missing a description in the init function docstring, which was added manually. The rest are stylistic changes triggered by the formatter.
MobileNetV3Lite failed with a RuntimeError during the first backprop stage. The error was caused in the final sequential stage of the network by a version mismatch between the ReLU and Dropout layers. Setting
inplace=Falsein the Dropout final Dropout layer realigns the versions and enables backprop to function correctly.The remaining changesets are required to adhere to the pre-commit formatting rules. The
activation_layerargument inMobileNetV3was missing a description in the init function docstring, which was added manually. The rest are stylistic changes triggered by the formatter.Fixes #58