dllllb / pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision
Apache License 2.0
216 stars 47 forks source link

Default optimizers and schedulers #16

Open ivkireev86 opened 2 years ago

ivkireev86 commented 2 years ago

Предусмотреть возможность задать оптимайзер и шедулер текстом. Например:

    optimizer_partial='adam',
    optimizer_adam_lr=0.001,
    lr_scheduler_partial='none',

Сделать эти значения дефолтными. Так можно будет не задавать их каждый раз. Примерный сценарий такой:

  1. Запустить все по дефолту
  2. Подтюнить optimizer_adam_lr
  3. Добавить lr_scheduler_partial
  4. Подтюнить optimizer_partial и lr_scheduler_partial
ivkireev86 commented 2 years ago

Similar #41