semipy.tools
Warning
This section is in construction.
This module provides useful tools for the proper functioning of the library.
Classes
| API Reference | Description |
|---|---|
| semipy.tools.SSLTrainer | Used to create a Train/Validation/Test 'environment' without PyTorch Lightning. |
Functions
| API Reference | Descritpion |
|---|---|
| semipy.tools.get_config | To read parameters from a YAML file. |
| semipy.tools.get_optimizer | To get the right optimizer with the correct parameters from torch.optim. |
| semipy.tools.get_cosine_schedule_with_warmup | To get a cosine scheduler. |