does not work for meta learning model

nmhkahn / torchsummaryX

torchsummaryX: Improved visualization tool of torchsummary

302 stars 32 forks source link

Closed vainaixr closed 5 years ago

vainaixr commented 5 years ago

input is of size (25, 3, 84, 84)

25 RGB images of size 84x84

nmhkahn commented 5 years ago

Hi, could you elaborate on your problem and your code?

vainaixr commented 5 years ago

hello, thanks for your reply.

it was my mistake, it is working now.

Screenshot (186)