update for running

This commit is contained in:
Xintao
2021-08-08 15:27:55 +08:00
parent 064df9956b
commit bef5e3cabd
2 changed files with 5 additions and 5 deletions

View File

@@ -2,10 +2,10 @@
import os.path as osp
from basicsr.train import train_pipeline
from .archs import *
from .data import *
from .models import *
import realesrgan.archs
import realesrgan.data
import realesrgan.models
if __name__ == '__main__':
root_path = osp.abspath(osp.join(__file__, osp.pardir))
root_path = osp.abspath(osp.join(__file__, osp.pardir, osp.pardir))
train_pipeline(root_path)