Feat: add base folder

boostcampaitech6 · Feb 7, 2024 · 34442b6 · 34442b6
1 parent 3059860
commit 34442b6
Show file tree

Hide file tree

Showing 11 changed files with 15 additions and 15 deletions.
diff --git a/src/args.py → src/base/args.py b/src/args.py → src/base/args.py
diff --git a/src/datasets.py → src/base/datasets.py b/src/datasets.py → src/base/datasets.py
diff --git a/src/models.py → src/base/models.py b/src/models.py → src/base/models.py
diff --git a/src/modules.py → src/base/modules.py b/src/modules.py → src/base/modules.py
diff --git a/src/preprocessing.py → src/base/preprocessing.py b/src/preprocessing.py → src/base/preprocessing.py
diff --git a/src/sample_submission.ipynb → src/base/sample_submission.ipynb b/src/sample_submission.ipynb → src/base/sample_submission.ipynb
diff --git a/src/trainers.py → src/base/trainers.py b/src/trainers.py → src/base/trainers.py
diff --git a/src/utils.py → src/base/utils.py b/src/utils.py → src/base/utils.py
diff --git a/src/inference.py b/src/inference.py
@@ -2,12 +2,12 @@
 
 import torch
 from torch.utils.data import DataLoader, SequentialSampler
-from args import parse_args
+from base.args import parse_args
 
-from datasets import SASRecDataset
-from models import S3RecModel
-from trainers import FinetuneTrainer
-from utils import (
+from base.datasets import SASRecDataset
+from base.models import S3RecModel
+from base.trainers import FinetuneTrainer
+from base.utils import (
     check_path,
     generate_submission_file,
     get_item2attribute_json,

diff --git a/src/run_pretrain.py b/src/run_pretrain.py
@@ -3,12 +3,12 @@
 import numpy as np
 import torch
 from torch.utils.data import DataLoader, RandomSampler
-from args import parse_args
+from base.args import parse_args
 
-from datasets import PretrainDataset
-from models import S3RecModel
-from trainers import PretrainTrainer
-from utils import (
+from base.datasets import PretrainDataset
+from base.models import S3RecModel
+from base.trainers import PretrainTrainer
+from base.utils import (
     EarlyStopping,
     check_path,
     get_item2attribute_json,

diff --git a/src/run_train.py b/src/run_train.py
@@ -3,12 +3,12 @@
 import numpy as np
 import torch
 from torch.utils.data import DataLoader, RandomSampler, SequentialSampler
-from args import parse_args
+from base.args import parse_args
 
-from datasets import SASRecDataset
-from models import S3RecModel
-from trainers import FinetuneTrainer
-from utils import (
+from base.datasets import SASRecDataset
+from base.models import S3RecModel
+from base.trainers import FinetuneTrainer
+from base.utils import (
     EarlyStopping,
     check_path,
     get_item2attribute_json,