From 31f3bff0ebf67d5e7e02a4b6a95cbd922fdea3a4 Mon Sep 17 00:00:00 2001 From: dsyoon Date: Tue, 9 Aug 2022 00:37:30 +0900 Subject: [PATCH] init --- VitTrainer.py | 4 ++-- stock/util/StockPredictor.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/VitTrainer.py b/VitTrainer.py index 0850575..a9809c5 100755 --- a/VitTrainer.py +++ b/VitTrainer.py @@ -40,7 +40,7 @@ class VitTrainer: self.label2id = {'none': 0, 'sell': 1, 'buy': 2} self.args = TrainingArguments( - f"stock_vit_predictor", + os.path.join(self.RESOURCE_PATH, 'model', f"stock_vit_predictor"), save_strategy="epoch", evaluation_strategy="epoch", learning_rate=2e-5, @@ -49,7 +49,7 @@ class VitTrainer: weight_decay=0.01, load_best_model_at_end=True, metric_for_best_model="accuracy", - logging_dir='logs', + logging_dir=os.path.join(self.RESOURCE_PATH, 'model', 'logs'), remove_unused_columns=False, num_train_epochs=14, ) diff --git a/stock/util/StockPredictor.py b/stock/util/StockPredictor.py index 06c8255..b1c2a5f 100644 --- a/stock/util/StockPredictor.py +++ b/stock/util/StockPredictor.py @@ -66,7 +66,7 @@ class StockPredictor: id2label=self.id2label, label2id=self.label2id) args = TrainingArguments( - f"stock_vit_predictor", + os.path.join(self.RESOURCE_PATH, 'model', f"stock_vit_predictor"), save_strategy="epoch", evaluation_strategy="epoch", learning_rate=2e-5, @@ -75,7 +75,7 @@ class StockPredictor: weight_decay=0.01, load_best_model_at_end=True, metric_for_best_model="accuracy", - logging_dir='logs', + logging_dir=os.path.join(self.RESOURCE_PATH, 'model', 'logs'), remove_unused_columns=False, num_train_epochs=4, )