* new usage of amp training.

* change the usage of amp and pure fp16 training.

* modified code as reviews