Cédric Deltheil
|
46dd710076
|
test_converter: use proper exception type
Follow up of #102
|
2023-10-18 14:39:24 +02:00 |
|
Benjamin Trom
|
6ddd901767
|
improve image_to_tensor and tensor_to_image utils
|
2023-10-17 18:08:58 +02:00 |
|
Benjamin Trom
|
0024191c58
|
improve debug print for chains
|
2023-10-10 15:25:09 +02:00 |
|
Benjamin Trom
|
a663375dc7
|
prevent setattr pytorch module to register on the Chain class
|
2023-10-10 14:46:15 +02:00 |
|
Cédric Deltheil
|
05126c8f4d
|
make gaussian_blur work with float16
|
2023-10-07 21:48:38 +02:00 |
|
Cédric Deltheil
|
665bcdc95c
|
add unit tests covering fluxion's gaussian_blur
|
2023-10-05 16:30:27 +02:00 |
|
Cédric Deltheil
|
d72e1d3478
|
chain: add insert_before_type
|
2023-09-25 13:54:26 +02:00 |
|
Pierre Chapuis
|
d54a38ae07
|
do not hardcode a CUDA device in tests
|
2023-09-06 19:33:48 +02:00 |
|
limiteinductive
|
88efa117bf
|
fix model comparison with custom layers
|
2023-09-05 12:34:38 +02:00 |
|
Pierre Chapuis
|
d311f779c0
|
test all chain manipulation methods
|
2023-08-23 17:49:59 +02:00 |
|
Pierre Chapuis
|
beacfe816b
|
reordering (match chain.py order)
|
2023-08-23 17:49:59 +02:00 |
|
Pierre Chapuis
|
e05c410a86
|
split test in two
|
2023-08-23 17:49:59 +02:00 |
|
Pierre Chapuis
|
337d2aea58
|
cosmetics
|
2023-08-23 17:49:59 +02:00 |
|
Pierre Chapuis
|
1065dfe10b
|
add empty __init__.py files to make pytest happy
(otherwise it wants unique file basenames)
|
2023-08-23 17:49:59 +02:00 |
|
Pierre Chapuis
|
a0c70ba7aa
|
add a test for StopIteration in walk
|
2023-08-23 12:15:56 +02:00 |
|
Pierre Chapuis
|
dec0d64432
|
make walk and layers not recurse by default
There is now a parameter to get the old (recursive) behavior.
|
2023-08-23 12:15:56 +02:00 |
|
Benjamin Trom
|
8c7298f8cc
|
fix chain slicing with structural copy
|
2023-08-22 11:44:11 +02:00 |
|
limiteinductive
|
108fa8f26a
|
add converter layer + tests
|
2023-08-21 12:09:58 +02:00 |
|
Cédric Deltheil
|
48f674c433
|
initial commit
|
2023-08-04 15:28:41 +02:00 |
|