Fixed embarrassing indentation problem

This commit is contained in:
Matthew Carrigan
2019-03-20 15:42:57 +00:00
parent 0540d360f2
commit c64c2fc4c2

View File

@@ -241,8 +241,7 @@ def main():
from apex.optimizers import FP16_Optimizer from apex.optimizers import FP16_Optimizer
from apex.optimizers import FusedAdam from apex.optimizers import FusedAdam
except ImportError: except ImportError:
raise ImportError( raise ImportError("Please install apex from https://www.github.com/nvidia/apex to use distributed and fp16 training.")
"Please install apex from https://www.github.com/nvidia/apex to use distributed and fp16 training.")
optimizer = FusedAdam(optimizer_grouped_parameters, optimizer = FusedAdam(optimizer_grouped_parameters,
lr=args.learning_rate, lr=args.learning_rate,