Update model params

testing update

Hook up model network instantiation and training function call

Add torchvision req

Add first model