Browse Source

Update encoder.py

main
BBing 2 years ago
parent
commit
37d1c3ad02
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      models/encoder.py

+ 1
- 0
models/encoder.py View File

@@ -2,6 +2,7 @@ import torch
import torch.nn as nn
import torch.nn.functional as F


class ConvLayer(nn.Module):
def __init__(self, c_in):
super(ConvLayer, self).__init__()


Loading…
Cancel
Save