Laurent Mazare b7cd58473b TinyViT backbone for segment-anything. (#787) ...

* TinyViT.

* More TinyViT.

* Add more to the tinyvit backbone.

* Proper padding.

* Plus ViT.

* Add the tiniest vit spec.

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

..

activation.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00

batch_norm.rs

VarBuilder cleanup (#627)

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

conv.rs

TinyViT backbone for segment-anything. (#787)

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

embedding.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +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

Softmax implementation for cuda. (#747)

2023-09-05 18:38:03 +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