serezd / scheduling_utils

LR Schedulers with C++ extensions

Implements some LR scheduling functions to decay or ramp values across timestamps.

Install

Install with: pip install scheduling_utils

Usage

Each scheduler is available with full python code or c++ code (ctypes is used for binding) wrapped into python classes.
To initialize and use for example a cosine scheduler, do the following:

# import the c++ cosine scheduler
from schedulers_cpp import CosineScheduler

# use this to use the full python version
# from schedulers import CosineScheduler

start_step = 0
stop_step = 10

start_value = 5
stop_value = 20

scheduler = CosineScheduler(start_step, stop_step, start_value, stop_value)

# get values corresponding to step
for step in range(10):
    value = scheduler.step(step)

Available Schedulers

Available Schedulers at current version:

• Linear:

  

• Cosine:

  

• LinearCosine:

  

scheduling_utils's Issues

OSError: [WinError 193] %1 не является приложением Win32

Traceback (most recent call last):
  File "C:\Users\*****\*****\vqvae-vqgan-pytorch-lightning-master\vqvae\train.py", line 15, in <module>
    from vqvae.model import VQVAE
  File "C:\Users\*****\*****\vqvae-vqgan-pytorch-lightning-master\vqvae\model.py", line 6, in <module>
    from scheduling_utils.schedulers_cpp import LinearScheduler, CosineScheduler, LinearCosineScheduler
  File "C:\Users\*****\*****\*****\venv\lib\site-packages\scheduling_utils\schedulers_cpp.py", line 6, in <module>
    schedulers_cpp = ctypes.CDLL(f'{Path(__file__).parent}\\cpp_extensions\\schedulers.so')
  File "C:\Users\*****\AppData\Local\Programs\Python\Python310\lib\ctypes\__init__.py", line 374, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: [WinError 193] %1 не является приложением Win32

Feels like DLL is 32-bit