Minor fixes

This commit is contained in:
pkufool 2021-12-10 15:30:00 +08:00
parent db924dcef5
commit 6dec4b2d8a
2 changed files with 2 additions and 2 deletions

View File

@ -878,7 +878,7 @@ class ConvolutionModule(nn.Module):
groups=channels,
bias=bias,
)
if self.use_bathnorm:
if self.use_batchnorm:
self.norm = nn.BatchNorm1d(channels)
self.pointwise_conv2 = nn.Conv1d(
channels,

View File

@ -15,7 +15,7 @@
# limitations under the License.
import math
from typing import Dict, List, Optional, Tuple
from typing import Dict, List, Optional, Tuple, Union
import torch
import torch.nn as nn