Add SequentialTransform

luigibonati / mlcolvar

A unified framework for machine learning collective variables for enhanced sampling simulations

MIT License

89 stars 24 forks source link

Add SequentialTransform #147

Closed EnricoTrizio closed 4 months ago

EnricoTrizio commented 4 months ago

Description

This PR adds a SequentialTransform helper class that allows combining multiple trasforms in a sequential way to fix #140. The new class works exactly like torch.nn.Sequential but also has in_features and out_features attributes.

Status

[x] Ready to go

codecov[bot] commented 4 months ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 92.56%. Comparing base (d9640bd) to head (d12bf39).

Additional details and impacted files