Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

pip_package_creation_second_attempt #3773

Merged
merged 20 commits into from
Jun 26, 2021
Merged

pip_package_creation_second_attempt #3773

Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
d843153
initial commit
SkalskiP Jun 25, 2021
99e3898
setup.py added
SkalskiP Jun 25, 2021
bfcc3c5
sys append fixes
SkalskiP Jun 25, 2021
f612822
sys append fixes 2
SkalskiP Jun 25, 2021
41f7d7a
yolov5_in_syspath
SkalskiP Jun 25, 2021
7b7cc21
move data inside package directory
SkalskiP Jun 25, 2021
5f5232f
move data inside package directory
SkalskiP Jun 25, 2021
73139ca
up
SkalskiP Jun 25, 2021
0bad8c7
test
SkalskiP Jun 25, 2021
4a4c8a5
test fix
SkalskiP Jun 25, 2021
84f1a81
test fix 2
SkalskiP Jun 25, 2021
320fdcd
test fix 3
SkalskiP Jun 25, 2021
9fe51b1
test fix 3
SkalskiP Jun 25, 2021
3518d49
test fix 4
SkalskiP Jun 25, 2021
251d2a5
test fix 5
SkalskiP Jun 25, 2021
9094dcd
test fix 6
SkalskiP Jun 25, 2021
af9eca1
test fix 7
SkalskiP Jun 25, 2021
0eb9561
more changes
SkalskiP Jun 25, 2021
c112693
more changes 2
SkalskiP Jun 26, 2021
7dae157
more changes 3
SkalskiP Jun 26, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
more changes
  • Loading branch information
@SkalskiP
SkalskiP committed Jun 25, 2021
commit 0eb95618cdaf9efef386d67f71956e2c4ed1dbed
8 changes: 5 additions & 3 deletions yolov5/utils/general.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -634,9 +634,10 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
return output


def strip_optimizer(f='best.pt', s=''): # from utils.general import *; strip_optimizer()
def strip_optimizer(f='best.pt', s=''):
# Strip optimizer from 'f' to finalize training, optionally save as 's'
x = torch.load(f, map_location=torch.device('cpu'))
with yolov5_in_syspath():
x = torch.load(f, map_location=torch.device('cpu'))
if x.get('ema'):
x['model'] = x['ema'] # replace model with ema
for k in 'optimizer', 'training_results', 'wandb_id', 'ema', 'updates': # keys
Expand All @@ -645,7 +646,8 @@ def strip_optimizer(f='best.pt', s=''): # from utils.general import *; strip_op
x['model'].half() # to FP16
for p in x['model'].parameters():
p.requires_grad = False
torch.save(x, s or f)
with yolov5_in_syspath():
torch.save(x, s or f)
mb = os.path.getsize(s or f) / 1E6 # filesize
print(f"Optimizer stripped from {f},{(' saved as %s,' % s) if s else ''} {mb:.1f}MB")

Expand Down