WIP: Lightning
4 unresolved threads
4 unresolved threads
Compare changes
- Tiago de Freitas Pereira authored
@@ -183,7 +183,12 @@ class IResNet(nn.Module):
Working on a training mechanism using pytorch lightning
Still a WIP