sadimanna/model_create.py

Created January 6, 2021 06:33

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/sadimanna/9bd112c771f8a80bd18fb0055eccebf6.js"></script>
Save sadimanna/9bd112c771f8a80bd18fb0055eccebf6 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

model_create.py

	model = new_resnet('resnet50','layer4',Bottleneck, [3, 4, 6, 3],True,True)
	model = model.to('cuda:0')

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment