jeta

jeta copied to clipboard

Add CNP

6

closes #25 ### Example: ```python from cnp import CNP, Aggregator class Encoder(linen.Module): @linen.compact def __call__(self, x, y, train=True): # x : (n_classes * shots, 20) # y : (n_classes *...

AnirudhM1

tests for maml and opti

1

Riddhi2587

logging support added

2

harini-si

veds12

MAML is expecting more parameters than being passed

1

https://github.com/SforAiDl/jeta/blob/e55bf83c9f89c662872f50a1ef8885c1085403ef/jeta/maml.py#L14-L15 The above lines cause an error while running `MAML` using [OptiTrainer](https://github.com/SforAiDl/jeta/blob/e55bf83c9f89c662872f50a1ef8885c1085403ef/jeta/opti_trainer.py) --- `OptiTrainer` calls the function `maml_adapt` and it's parameters are passed in `OptiTrainer` itself. > Note: `maml_adapt` is...

AnirudhM1

Add Logging Support

5

Add: - [Wandb](https://wandb.ai/site) logger - [Tensorboard](https://www.tensorflow.org/tensorboard/get_started) logger - Local logger

veds12

Add tests for algorithms

2

Currently, there are no tests, and hence no code coverage. We should test the algorithms (at least 1 or 2 epochs with a small fas) to check if the forward...

abhi-glitchhg

Extend MAML to support FO-MAML and i-MAML

3

[1] (FO-MAML) https://arxiv.org/abs/1803.02999 [2] (i-MAML) https://papers.nips.cc/paper/2019/file/072b030ba126b2f4b2374f342be9ed44-Paper.pdf

veds12

Add Conditional Neural Processes (CNP)

2

Paper: https://arxiv.org/abs/1807.01613

veds12

veds12