[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
026ad463a9
commit
0f706ce543
|
@ -285,7 +285,7 @@ def add_actor_information_and_train(
|
||||||
# ds_meta=cfg.dataset,
|
# ds_meta=cfg.dataset,
|
||||||
env_cfg=cfg.env,
|
env_cfg=cfg.env,
|
||||||
)
|
)
|
||||||
|
|
||||||
assert isinstance(policy, nn.Module)
|
assert isinstance(policy, nn.Module)
|
||||||
policy.train()
|
policy.train()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue