Spaces:
Running
Running
set seed
Browse files
config.py
CHANGED
|
@@ -20,6 +20,7 @@ SMELLS = [
|
|
| 20 |
|
| 21 |
|
| 22 |
# Configurations
|
|
|
|
| 23 |
DROPOUT_PROB = 0.1
|
| 24 |
HIDDEN_SIZE = 768
|
| 25 |
NUM_LABELS = 9
|
|
@@ -29,4 +30,7 @@ PRETRAINED_MODEL = "microsoft/graphcodebert-base"
|
|
| 29 |
MODEL_SAVE_DIR = Path(__file__).parents[1] / "models_weights"
|
| 30 |
WEIGHTS_FILE_NAME = "weights.bin"
|
| 31 |
MODEL_SAVE_PATH = MODEL_SAVE_DIR / WEIGHTS_FILE_NAME
|
| 32 |
-
HF_REPO_NAME = f"mjavadmt/EnergySmell"
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
# Configurations
|
| 23 |
+
SEED_NUM = 42
|
| 24 |
DROPOUT_PROB = 0.1
|
| 25 |
HIDDEN_SIZE = 768
|
| 26 |
NUM_LABELS = 9
|
|
|
|
| 30 |
MODEL_SAVE_DIR = Path(__file__).parents[1] / "models_weights"
|
| 31 |
WEIGHTS_FILE_NAME = "weights.bin"
|
| 32 |
MODEL_SAVE_PATH = MODEL_SAVE_DIR / WEIGHTS_FILE_NAME
|
| 33 |
+
HF_REPO_NAME = f"mjavadmt/EnergySmell"
|
| 34 |
+
|
| 35 |
+
torch.manual_seed(SEED_NUM)
|
| 36 |
+
set_seed(SEED_NUM)
|