Fix

Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
NVIDIA · Oct 24, 2024 · e4a536e · e4a536e
1 parent df2bdac
commit e4a536e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/docs/examples/use_cases/paddle/resnet50/utils/config.py b/docs/examples/use_cases/paddle/resnet50/utils/config.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import os
+import sys
 import copy
 import argparse
 import logging
@@ -23,6 +24,8 @@
 
 _AUTO_LAST_EPOCH = "auto"
 
+_true_set = {'yes', 'true', 't', 'y', '1'}
+_false_set = {'no', 'false', 'f', 'n', '0'}
 
 # based on https://github.com/symonsoft/str2bool/tree/master
 # BSD3 license