60a61
>         print(f"kwargs: {kwargs}")
62a64
>         print(f"device: {self.device}")
129c131
<             layer='penultimate')
---
>             layer='penultimate', device=self.device)
224a227
>         print(f"self.device: {self.device}")