Update sample to use lightningmodule

update README since old custom impl has been deleted

replace custom impl of train with LightningModule

include input file

update readme to have the proper dir name

device to cpu on sample.py

init