diff --git a/examples/run_squad.py b/examples/run_squad.py index 0e9aec81..e8ed71cd 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -917,7 +917,7 @@ def main(): if args.fp16: try: - from apex.optimizer import FP16_Optimizer + from apex.optimizers import FP16_Optimizer from apex.optimizers import FusedAdam except ImportError: raise ImportError("Please install apex from https://www.github.com/nvidia/apex to use distributed and fp16 training.")