From 04f78fbcbc097dd4cc42cd60e8842d0d804afb62 Mon Sep 17 00:00:00 2001
From: rubberhead <bjchenzhengyi@hotmail.com>
Date: Wed, 28 Feb 2024 15:28:38 +0000
Subject: [PATCH] maintenance

---
 model/glue.py | 9 +++++++++
 train-dist.py | 0
 2 files changed, 9 insertions(+)
 create mode 100644 model/glue.py
 create mode 100644 train-dist.py

diff --git a/model/glue.py b/model/glue.py
new file mode 100644
index 0000000..afdbb21
--- /dev/null
+++ b/model/glue.py
@@ -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
diff --git a/train-dist.py b/train-dist.py
new file mode 100644
index 0000000..e69de29