For 3D Networks?

[DISAPPEARED13]

Hello! I want to check my 3D CNN model's GPU using with torchstat, but it seems not support inputs with more than 3D... assertionerror occured.. any solutions to solve this problem? or can you recommend some effective tools to check GPU usage? thx..

[DISAPPEARED13]

assert isinstance(input_size, (tuple, list)) and len(input_size) == 3

[DISAPPEARED13]

if i cancel the assert sentence, torch.nn 's 3D layers are not suported by torchstat

[ShinoSpace]

try another package "thop"