Ross Wightman 2180800646 MQA query_strides bugs fix #2237. No padding for avg_pool2d if not 'same', use scale_factor for Upsample.

2024-07-19 14:26:54 -07:00

..

data

Fix pass through of input / target keys so ImageDataset readers so args work with hfds instead of just hfids (iterable)

2024-07-17 10:11:46 -07:00

layers

MQA query_strides bugs fix #2237. No padding for avg_pool2d if not 'same', use scale_factor for Upsample.

2024-07-19 14:26:54 -07:00

loss

…

models

eva.py: fixed bug in applying attention mask

2024-07-19 15:12:04 +03:30

optim

…

scheduler

…

utils

…

__init__.py

…

version.py

…