ppgbp modifications #13

Francesco-Carlucci · 2025-05-05T20:29:45Z

Adding all modifications needed to support the ppgbp architectures, ResNet and UNet like.

getitem operation support
- adding 'features_slicing' in meta
- adding correct_get_item to fix indices during exports
- adding GetItemFeaturesCalculator
- tests on group conv with any number of groups
- test with any possible indexing method (slice, list, integer, boolean mask and torch tensor
Adding PITInstanceNorm and PITPReLU classes
adding minor propagting layers (sigmoid, upsample...)
fix group conv in supernet, using index instead of name
adding None value in ModAttrFeaturesCalculator, so that GetItemFeaturesCalculator works with MPS, where there are no masks
adding group convolution notebook in examples/

…PER_LAYER

…ersion

…olution in supernet

Francesco-Carlucci added 13 commits March 12, 2025 15:05

minor fixes to avoid tests crash

f8d097c

adding ppgbp resnet and unet models for tests

a148d8b

adding pit tests on unet, resnet and group conv

8cdecd8

adding group convolution and indexing toy models for tests

4713dea

adding is_feature_get_item and minor propagating layers

6a8108a

adding feature_slicing to node meta

d28d67c

adding GetItemFeaturesCalculator and ModAttr mask None value for MPS …

90d6d6c

…PER_LAYER

adding correct get item function, PITInstanceNorm1d and PITPReLU conv…

faaab53

…ersion

adding instancenorm1d and prelu pit files

b79806d

modifying export graph to work with best index too, to use group conv…

33dd3cd

…olution in supernet

Test MPS PER_LAYER conversion of a model using indexing, masks are None

0085feb

Merge branch 'main' into ppgbp_modifications

e0e05a4

adding group convolutions notebook

eaeff7a

Provide feedback