Equation attributes not moved to device #688
Description
Describe the bug
The attributes of any class inheriting from Equation are not automatically moved to the correct training device.
{{ message }}
Describe the bug
The attributes of any class inheriting from Equation are not automatically moved to the correct training device.