zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

comment

Browse source
This commit is contained in:
Mouleesh 2024-03-13 16:30:34 +00:00
parent 60fc7b7bd1
commit bb4a4ee0d6
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
network/glue.py
View file

@ -1,6 +1,6 @@
# Glue layer for transforming whole pictures into 384x384 sequence for encoder input # Glue layer for transforming whole pictures into 384x384 sequence for encoder input
from dataclasses import dataclass from dataclasses import dataclass
from itertools import product from itertools import product # product library
import torch import torch
import torch.nn as nn import torch.nn as nn