Fix long time step() detach
This commit is contained in:
parent
9634ab12c5
commit
ef3868066e
1 changed files with 0 additions and 1 deletions
|
|
@ -52,7 +52,6 @@ class Module(ABC):
|
|||
|
||||
def to(self, device):
|
||||
for _, _, parameter in self.parameters():
|
||||
print(type(parameter))
|
||||
parameter.to(device)
|
||||
|
||||
return self
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue