Update trainDistilBERT.py

trainDistilBERT.py

@@ -10,13 +10,11 @@ from transformers.modeling_outputs import SequenceClassifierOutput
 from datasets import load_dataset, Dataset
 import pandas as pd
 from sklearn.metrics import accuracy_score, precision_recall_fscore_support, confusion_matrix, roc_auc_score
-import wandb
 import huggingface_hub
 import os
 import random
 import numpy as np
 
-WANDB_MODE="offline"
 
 config = {
 	"mdl": "distilbert-base-uncased",
@@ -33,7 +31,7 @@ config = {
 	"epsilon": 1e-8
 }
 
-wandb.init(project="MarkIII_ACARIS", entity="simtoonia", config=config)
+#wandb.init(project="MarkIII_ACARIS", entity="simtoonia", config=config)
 
 
 def lockSeed(seed):