diff --git a/.gitignore b/.gitignore
index 5c93fc12348f..d49327f5e32d 100755
--- a/.gitignore
+++ b/.gitignore
@@ -249,4 +249,6 @@ crashlytics.properties
 crashlytics-build.properties
 fabric.properties
 
-supervisely_lib
\ No newline at end of file
+supervisely_lib
+
+app_debug_data/
\ No newline at end of file
diff --git a/.vscode/launch.json b/.vscode/launch.json
index 7cb46d0c698d..5a677e2427a7 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -5,7 +5,7 @@
             "name": "Debug",
             "type": "python",
             "request": "launch",
-            "program": "${workspaceFolder}/supervisely/serve/src/main.py",
+            "program": "${workspaceFolder}/supervisely/train/src/sly_train.py",
             "console": "integratedTerminal",
             "justMyCode": false,
             "env": {
diff --git a/requirements.txt b/requirements.txt
index 5c9d3f83ccad..4d365d73fa41 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,6 +4,10 @@ supervisely==6.72.12
 opencv-python-headless==4.5.5.62
 opencv-python==4.5.5.62
 
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch==1.11.0
+torchvision==0.12.0
+
 # base ----------------------------------------
 PyYAML==5.4.1
 tensorboard==2.4.1
@@ -15,8 +19,8 @@ seaborn==0.11.1
 coremltools==4.1
 onnx>=1.8.0
 onnxruntime==1.8.0
-# scikit-learn==1.1.1  # for coreml quantization
-numpy==1.19.0
+scikit-learn==1.1.1  # for coreml quantization
+numpy==1.21.2
 # extras --------------------------------------
 thop==0.0.31-2005241907  # FLOPS computation
 pycocotools==2.0.2  # COCO mAP
diff --git a/train.py b/train.py
index eb585d3c949b..a46c8bf16188 100644
--- a/train.py
+++ b/train.py
@@ -67,6 +67,7 @@
 
 
 def train(hyp, opt, device, tb_writer=None):
+
     train_batches_uploaded = False
 
     logger.info("hyperparameters", extra=hyp)