Update delta-iris/src/models/utils.py
Browse files
delta-iris/src/models/utils.py
CHANGED
|
@@ -13,7 +13,7 @@ import torch.nn as nn
|
|
| 13 |
import torch.nn.functional as F
|
| 14 |
from torch.optim import AdamW
|
| 15 |
|
| 16 |
-
from .data import Episode
|
| 17 |
|
| 18 |
|
| 19 |
def configure_optimizer(model: nn.Module, learning_rate: float, weight_decay: float, *blacklist_module_names) -> AdamW:
|
|
|
|
| 13 |
import torch.nn.functional as F
|
| 14 |
from torch.optim import AdamW
|
| 15 |
|
| 16 |
+
from src.data import Episode
|
| 17 |
|
| 18 |
|
| 19 |
def configure_optimizer(model: nn.Module, learning_rate: float, weight_decay: float, *blacklist_module_names) -> AdamW:
|