src/sagemaker_defect_detection/detector.py (5 lines): - line 275: # TODO: scores are predictions scores, not a metric! iou? + acc? - line 290: # TODO: above - line 373: # TODO: `model_fn` doesn't get more args - line 395: # TODO: add deterministic training - line 437: # precision=16 if args.use_16bit else 32, # TODO: apex src/sagemaker_defect_detection/classifier.py (3 lines): - line 217: # TODO: `model_fn` doesn't get more args - line 236: # TODO: add deterministic training - line 256: # precision=16 if args.use_16bit else 32, # TODO: amp apex support src/sagemaker_defect_detection/dataset/neu.py (1 line): - line 61: # TODO: add split ratios as parameters