ShaswatRobotics
/

world_model_test

ShaswatRobotics commited on 13 days ago

Commit

5553f7f

verified ·

1 Parent(s): 69c71b7

Update delta-iris/src/models/utils.py

Files changed (1) hide show

delta-iris/src/models/utils.py CHANGED Viewed

@@ -13,7 +13,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 from torch.optim import AdamW
-from .data import Episode
 def configure_optimizer(model: nn.Module, learning_rate: float, weight_decay: float, *blacklist_module_names) -> AdamW:

 import torch.nn.functional as F
 from torch.optim import AdamW
+from src.data import Episode
 def configure_optimizer(model: nn.Module, learning_rate: float, weight_decay: float, *blacklist_module_names) -> AdamW: