Method__init__(self, n_embd, n_layer, layer_id, shift_mode='q_shift',
channel_gamma=1/4, shift_pixel=1, ini
model/modules/urwkv.py:125
Method__init__(self, n_embd, n_layer, layer_id, shift_mode='q_shift',
channel_gamma=1/4, shift_pixel=1, hid
model/modules/urwkv.py:246
Method__init__(self, optimizer, start_lr=0.03, end_lr=5e-4,
epochs=50, last_epoch=-1, **kwargs)
custom_utils/lr_scheduler.py:49
Method__init__(self, optimizer, start_lr=0.01, end_lr=None,
step=10, mult=0.1, epochs=50, last_epoch=-1, **
custom_utils/lr_scheduler.py:62
Method__init__(self, optimizer, start_lr=0.01, end_lr=None,
steps=[10, 20, 30, 40], mult=0.5, epochs=50,
custom_utils/lr_scheduler.py:78
Method__init__(self, optimizer, start_lr=0.01, end_lr=0.005,
epochs=50, last_epoch=-1, **kwargs)
custom_utils/lr_scheduler.py:104
Method__init__(self, optimizer, start_lr=0.01, end_lr=0.005,
epochs=50, last_epoch=-1, **kwargs)
custom_utils/lr_scheduler.py:113