Skip to content

Update pruning.py#1

Open
Pawan-KS wants to merge 1 commit into
AryanRaj315:masterfrom
Pawan-KS:master
Open

Update pruning.py#1
Pawan-KS wants to merge 1 commit into
AryanRaj315:masterfrom
Pawan-KS:master

Conversation

@Pawan-KS
Copy link
Copy Markdown
Collaborator

Changed 54th line in pruning.py file.
So now, uncommenting the 53rd and 54th line will load the pretrained weights to resnet152 model.
This is in consideration that the weight file (resnet152-b121ed2d.pth) is present in the parent directory of ChipNet.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant