BayesWatch / deep-kernel-transfer

Official pytorch implementation of the paper "Bayesian Meta-Learning for the Few-Shot Setting via Deep Kernels" (NeurIPS 2020)
https://arxiv.org/abs/1910.05199
201 stars 29 forks source link

MAML regression #6

Closed ZohrehAdabi closed 3 years ago

ZohrehAdabi commented 3 years ago

Hi @mpatacchiola, @elliotjcrowley

Could you provide the code for MAML regression on QMUL (head pose detection) please? Thanks in advancce.

mpatacchiola commented 3 years ago

It is the same code used for the classification case. For the regressor, you just need to replace the output softmax with a single linear unit.