misc
This commit is contained in:
parent
a1743d26aa
commit
b92ff8de78
|
@ -14,14 +14,6 @@ from trainer.networks import register_model
|
||||||
from utils.util import checkpoint
|
from utils.util import checkpoint
|
||||||
|
|
||||||
|
|
||||||
def is_latent(t):
|
|
||||||
return t.dtype == torch.float
|
|
||||||
|
|
||||||
|
|
||||||
def is_sequence(t):
|
|
||||||
return t.dtype == torch.long
|
|
||||||
|
|
||||||
|
|
||||||
class SubBlock(nn.Module):
|
class SubBlock(nn.Module):
|
||||||
def __init__(self, inp_dim, contraction_dim, heads, dropout):
|
def __init__(self, inp_dim, contraction_dim, heads, dropout):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user