search

kbressem / trainlib

MIT License
4 stars 0 forks source link

Shorten post transforms #35

Closed kbressem closed 1 year ago

kbressem commented 2 years ago

Use tuples with 

 get_transform(
            "AsDiscreted",
            config=config,
            keys=CommonKeys.PRED,
            argmax=True,
            to_onehot=config.model.out_channels,
            num_classes=config.model.out_channels,
        ),