3402 Add support to set other pickle related args

[Nic-Ma]

Fixes #3402 .

Description

This PR added support to set other pickle related args for torch.save, especially for big cache content as user requested in #3402 .

Status

Ready

Types of changes

• Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[Nic-Ma]

I added unit tests to cover the args, and also verified with the below program to cache big content with pickle_protocol=4:

train_transforms = Compose(
    [
        LoadImaged(keys=["image", "label"]),
        AddChanneld(keys=["image", "label"]),
        Spacingd(keys=["image", "label"], pixdim=(0.5, 0.5, 0.5), mode=("bilinear", "nearest")),
        RepeatChanneld(keys="image", repeats=3),
        EnsureTyped(keys=["image", "label"]),
    ]
)

The cached content size:

Thanks.

[Nic-Ma]

/black

[Nic-Ma]

/black

[Nic-Ma]

Some CI lib may be updated so the flake8 test failed in transform.py:

flake8
4.0.1 (flake8-bugbear: 21.11.28, flake8-comprehensions: 3.7.0,
flake8-executable: 2.1.1, flake8-pyi: 20.10.0, mccabe: 0.6.1, naming: 0.12.1,
pycodestyle: 2.8.0, pyflakes: 2.4.0) CPython 3.8.12 on Linux
/home/runner/work/MONAI/MONAI/monai/transforms/transform.py:207:5: B018 Found useless expression. Either assign it to a variable or remove it.
1     B018 Found useless expression. Either assign it to a variable or remove it.
1
Check failed!

I slightly adjusted the document of Transform class to avoid the flake8 issue.

Thanks.

[Nic-Ma]

/black

[Nic-Ma]

/build

[wyli]

Thanks, it looks good to me apart from the minor issue.

[Nic-Ma]

/build

[wyli]

/build