Browse Source

Update encoder.py

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

+ 1
- 1
models/encoder.py View File

@@ -1,7 +1,7 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
im


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


Loading…
Cancel
Save