Update dnaflash.py
Browse files- dnaflash.py +3 -0
dnaflash.py
CHANGED
|
@@ -13,6 +13,9 @@ import torch.utils.checkpoint
|
|
| 13 |
from torch import nn, Tensor
|
| 14 |
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
# helper functions
|
| 17 |
|
| 18 |
def exists(val):
|
|
|
|
| 13 |
from torch import nn, Tensor
|
| 14 |
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
| 15 |
|
| 16 |
+
from typing import Optional, Tuple, Union, Any
|
| 17 |
+
|
| 18 |
+
|
| 19 |
# helper functions
|
| 20 |
|
| 21 |
def exists(val):
|