From 354f21a300da09efe0b11ed533f91e8f63857740 Mon Sep 17 00:00:00 2001
From: toummHus <147893353+toummHus@users.noreply.github.com>
Date: Fri, 16 Aug 2024 20:12:29 +0800
Subject: [PATCH] Update train.py

---
 train.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/train.py b/train.py
index 7b9802c..dbdd3f5 100644
--- a/train.py
+++ b/train.py
@@ -8,7 +8,6 @@
 from torch.utils.data import DataLoader
 
 from utils.dataset_utils import TrainDataset
-from net.model import PromptIR
 from utils.schedulers import LinearWarmupCosineAnnealingLR
 import numpy as np
 import wandb