Fixed embarrassing indentation problem
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user