Laurent Mazare 2746f2c4be DiffNeXt/unet (#859) ...

* DiffNeXt/unet

* Start adding the vae.

* VAE residual block.

* VAE forward pass.

* Add pixel shuffling.

* Actually use pixel shuffling.

2023-09-15 10:14:02 +01:00

..

activation.rs

Add leaky-relu in the activation enum. (#858)

2023-09-15 07:05:38 +01:00

batch_norm.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

conv.rs

Add weight, bias, and hidden_size methods (#816)

2023-09-11 16:01:11 +01:00

embedding.rs

Add weight, bias, and hidden_size methods (#816)

2023-09-11 16:01:11 +01:00

func.rs

Add a couple functions required for yolo. (#527)

2023-08-20 17:02:05 +01:00

group_norm.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

init.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

layer_norm.rs

Segment Anything - process images (#766)

2023-09-07 19:22:45 +01:00

lib.rs

Expose the conv2d-transpose layers. (#761)

2023-09-07 06:04:52 +01:00

linear.rs

Segment Anything - process images (#766)

2023-09-07 19:22:45 +01:00

loss.rs

Fix typo in the nll function document (#742)

2023-09-05 09:25:11 +01:00

ops.rs

DiffNeXt/unet (#859)

2023-09-15 10:14:02 +01:00

optim.rs

Add the optimizer trait. (#702)

2023-09-01 12:55:39 +01:00

rnn.rs

Fix the rnn tests for accelerate. (#704)

2023-09-01 13:21:38 +01:00

var_builder.rs

Use an arc in the varbuilder rather than rc. (#757)

2023-09-06 15:29:09 +01:00

var_map.rs

Move the VarMap to a separate file (#525)

2023-08-20 14:25:07 +01:00