Anton Lozhkov d9316bf8bc Fix mutable proj_out weight in the Attention layer (#73) ...

* Catch unused params in DDP

* Fix proj_out, add test

2022-07-04 12:36:37 +02:00

..

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

test_layers_utils.py

add tests for 1D Up/Downsample blocks (#72)

2022-07-04 11:41:04 +02:00

test_modeling_utils.py

Fix mutable proj_out weight in the Attention layer (#73)

2022-07-04 12:36:37 +02:00

test_scheduler.py

image->sample in schedule tests

2022-06-17 15:51:33 +02:00