init (not checked bat script yet)
This commit is contained in:
@@ -4,9 +4,7 @@ from torch import nn
|
||||
from torch.nn import functional as F
|
||||
|
||||
import commons
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
from tools.log import logger as logging
|
||||
|
||||
|
||||
class LayerNorm(nn.Module):
|
||||
@@ -69,7 +67,7 @@ class Encoder(nn.Module):
|
||||
self.cond_layer_idx = (
|
||||
kwargs["cond_layer_idx"] if "cond_layer_idx" in kwargs else 2
|
||||
)
|
||||
logging.debug(self.gin_channels, self.cond_layer_idx)
|
||||
# logging.debug(self.gin_channels, self.cond_layer_idx)
|
||||
assert (
|
||||
self.cond_layer_idx < self.n_layers
|
||||
), "cond_layer_idx should be less than n_layers"
|
||||
|
||||
Reference in New Issue
Block a user