Merge pull request #197 from supervisely-ecosystem/test_tqdm

fix: AttributeError: 'tqdm_sly' object has no attribute 'iterable'

.gitignore

@@ -249,4 +249,6 @@ crashlytics.properties
 crashlytics-build.properties
 fabric.properties
 
-supervisely_lib
+supervisely_lib
+
+app_debug_data/

.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": {

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

train.py

@@ -67,6 +67,7 @@
 
 
 def train(hyp, opt, device, tb_writer=None):
+
     train_batches_uploaded = False
 
     logger.info("hyperparameters", extra=hyp)