Merge pull request #53 from shanooni/torch_pretrained_deprecation_warning_fix

fix: update readMe

fixed all the deprecated warnings

Fix VGG model tensor shape mismatch in copy_data function

Fix VGG model tensor shape mismatch in copy_data function

updated alexnet model to remove deprecation

upgraded test regarding resnet model updates

default test passing

Update README

Update README.md

Update README.md

Merge pull request #45 from cobanov/patch-2

Added gpu select feature

update pypi

Merge pull request #35 from yuliang0225/lshang