setup env

run_pytorch_nn.bat

@@ -5,6 +5,14 @@ echo ============================================================
 
 call conda activate gpt-gpu
 
+REM Install missing dependencies if needed
+echo Checking for required packages...
+python -c "import sklearn" 2>NUL
+if %ERRORLEVEL% NEQ 0 (
+    echo Installing scikit-learn...
+    call conda install -y scikit-learn
+)
+
 REM Parse command-line arguments
 set MODE=train
 set MODEL_TYPE=cnn