Open maxfreu opened 8 months ago
adapt for new pytorch lightning and pytorch version. trainer now gets accelerator kwarg or so, not gpus anymore.
adapt for new pytorch lightning and pytorch version. trainer now gets accelerator kwarg or so, not gpus anymore.