Hi team,
I have a pytorch model (.pth file) with me. Could you please help me to identify various layers in it.
Thank you
Nidheesh
Hi team,
I have a pytorch model (.pth file) with me. Could you please help me to identify various layers in it.
Thank you
Nidheesh
model = torch.load("model.pth", map_location='cpu')
layers = [module for module in model.modules()]