added wandb, fixed y_val serialisation error

main.py

@@ -56,7 +56,7 @@ def main(train=True, retrain=False):
                "balanced_accuracy_val": balanced_accuracy_val,
                "classification_report_val_table": classification_report_val_table,
                "predictions_val_table": predictions_val_table,
-               "y_val": y_val})
+               "y_val": y_val.tolist()})
     run.finish()
 
     # Save final predictions