search

sungyubkim / GBML

A collection of Gradient-Based Meta-Learning Algorithms with pytorch
MIT License
61 stars 8 forks source link

ResNet makes 'nan' loss in 'imaml' mode #8

Closed seonhoon1002 closed 1 year ago

seonhoon1002 commented 1 year ago

Thank you for sharing your codes.

image

python main.py --alg "iMAML" --net "ResNet"

I have some problem, when I train imaml with ResNet.

ConVNet case is Ok for me, but ResNet case makes some problem.

Did you ever face like this problem?