Small bug fixes/imports

This commit is contained in:
Daniel Povey 2022-03-02 16:34:55 +08:00
parent 9d1b4ae046
commit 9ed7d55a84

View File

@ -17,6 +17,8 @@
import torch
import torch.nn as nn
from torch import Tensor
from typing import Tuple
class Conv2dSubsampling(nn.Module):