search

LarsKue / lightning-trainable

A default trainable module for pytorch lightning.
MIT License
10 stars 1 forks source link

on_train_start does not transfer batch to self.device #16

Closed LarsKue closed 11 months ago