Skip to content
Discussion options

You must be logged in to vote

I figured it out. I had a mistake in my model generation where I was using a List for modules rather than a Pytorch class.

I changed:
self.fc = []
to
self.fc = nn.Sequential()

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by ddicostanzo
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
strategy: dp (removed in pl) DataParallel strategy: ddp DistributedDataParallel accelerator: cuda Compute Unified Device Architecture GPU pl Generic label for PyTorch Lightning package
1 participant