Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

[Export Refactor][Image Classification] create_deployment_folder function #1889

Merged
merged 26 commits into from
Dec 12, 2023
Merged

[Export Refactor][Image Classification] create_deployment_folder function #1889

Changes from 1 commit
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
6d8991c
initial commit
dbogunowicz Dec 5, 2023
059ceab
looking good, time to cleanup
dbogunowicz Dec 6, 2023
b390c2e
Delete src/sparseml/export/helpers.py
dbogunowicz Dec 6, 2023
c2c8444
Delete tests/sparseml/export/test_helpers.py
dbogunowicz Dec 6, 2023
6ce6ba5
ready for review
dbogunowicz Dec 6, 2023
6f3e5e7
Merge branch 'feature/damian/create_model_ic' of github.com:neuralmag…
dbogunowicz Dec 6, 2023
5dfbdcd
improve design
dbogunowicz Dec 6, 2023
042c193
tests pass
dbogunowicz Dec 6, 2023
29cfa1d
reuse _validate_dataset_num_classes
dbogunowicz Dec 6, 2023
ab73aec
initial commit
dbogunowicz Dec 6, 2023
f628532
Update src/sparseml/pytorch/image_classification/integration_helper_f…
dbogunowicz Dec 6, 2023
b93b634
Update src/sparseml/pytorch/image_classification/integration_helper_f…
dbogunowicz Dec 6, 2023
e7606cd
ready for review
dbogunowicz Dec 7, 2023
ea9cb61
Update src/sparseml/export/export.py
dbogunowicz Dec 7, 2023
9572e0b
Update src/sparseml/integration_helper_functions.py
dbogunowicz Dec 7, 2023
0229deb
initial commit
dbogunowicz Dec 7, 2023
a8c1b68
fixes
dbogunowicz Dec 7, 2023
2379354
ready for review
dbogunowicz Dec 7, 2023
741fb12
nit
dbogunowicz Dec 7, 2023
ebdeb9f
add return
dbogunowicz Dec 7, 2023
8b2fca0
initial commit
dbogunowicz Dec 7, 2023
ff52598
initial commit
dbogunowicz Dec 8, 2023
08f42c2
initial commit
dbogunowicz Dec 8, 2023
2951093
Merge remote-tracking branch 'origin/feature/damian/feature_branch_ex…
dbogunowicz Dec 11, 2023
c0dabe0
fix rebase, tests_work
dbogunowicz Dec 12, 2023
a4e6ea4
ready to push
dbogunowicz Dec 12, 2023
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
Update src/sparseml/pytorch/image_classification/integration_helper_f… 
…unctions.py
  • Loading branch information
@dbogunowicz
dbogunowicz authored Dec 6, 2023
commit b93b63405f94e6d18636f67a0e2d9db0e424760c
2 changes: 1 addition & 1 deletion src/sparseml/pytorch/image_classification/integration_helper_functions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def create_dummy_input(
if not validation_loader:
# create fake data for export
validation_loader = [[torch.randn(1, 3, image_size, image_size)]]
return next(iter(val_loader))[0]
return next(iter(validation_loader))[0]


@IntegrationHelperFunctions.register(name=Integrations.image_classification.value)
Expand Down