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

maintenance

Browse source
This commit is contained in:
Zhengyi Chen 2024-02-28 15:28:38 +00:00
parent 99266d9c92
commit 04f78fbcbc
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
model/glue.py Normal file
View file

@ -0,0 +1,9 @@
# Glue layer for transforming whole pictures into 384x384 sequence for encoder
# input
import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision
from torchvision import transforms
import numpy as np

0
train-dist.py Normal file
View file