migrate : examples to pytorch lightening

[kshitijrajsharma]

to keep it clean , pytorch lightening is great practice to make pipeline reproducible and understandable auto detects gpu and has nice callbacks , we can recommend this for pytorch based models too !