regroup files

2021-08-08 14:46:43 +08:00
parent f932289af1
commit 9baa0b3d00
9 changed files with 5 additions and 5 deletions
--- a/realesrgan/train.py
+++ b/realesrgan/train.py
@@ -0,0 +1,10 @@
+import os.path as osp
+from basicsr.train import train_pipeline
+
+import realesrgan.archs  # noqa: F401
+import realesrgan.data  # noqa: F401
+import realesrgan.models  # noqa: F401
+
+if __name__ == '__main__':
+    root_path = osp.abspath(osp.join(__file__, osp.pardir))
+    train_pipeline(root_path)