optional arguments: -h, --help show this help message and exit --exp_name EXP_NAME, -n EXP_NAME experiment name. A folder with this name will be created in the log_path. --epochs EPOCHS, -e EPOCHS number of epochs --batch_size BATCH_SIZE, -b BATCH_SIZE batch size --num_workers NUM_WORKERS, -w NUM_WORKERS number of DataLoader workers --lr LR, -l LR initial learning rate --disable_lr_decay disable learning rate decay --lr_decay_steps LR_DECAY_STEPS learning rate decay steps --lr_decay_gamma LR_DECAY_GAMMA learning rate decay gamma --optimizer OPTIMIZER, -o OPTIMIZER optimizer name. Currently, only `SGD` and `Adam` are supported. --weight_decay WEIGHT_DECAY weight decay --momentum MOMENTUM, -m MOMENTUM momentum --nesterov enable nesterov --pretrained_weight_path PRETRAINED_WEIGHT_PATH, -p PRETRAINED_WEIGHT_PATH pre-trained weight path. Weights will be loaded before training starts. --checkpoint_path CHECKPOINT_PATH, -c CHECKPOINT_PATH previous checkpoint path. Checkpoint will be loaded before training starts. It includes the model, the optimizer, the epoch, and other parameters. --log_path LOG_PATH log path. tensorboard logs and checkpoints will be saved here. --disable_tensorboard_log, -u disable tensorboard logging --model_c MODEL_C HRNet c parameter --model_nof_joints MODEL_NOF_JOINTS HRNet nof_joints parameter --model_bn_momentum MODEL_BN_MOMENTUM HRNet bn_momentum parameter --disable_flip_test_images disable image flip during evaluation --image_resolution IMAGE_RESOLUTION, -r IMAGE_RESOLUTION image resolution --coco_root_path COCO_ROOT_PATH COCO dataset root path --coco_bbox_path COCO_BBOX_PATH path of detected bboxes to use during evaluation --seed SEED, -s SEED seed --device DEVICE, -d DEVICE device
Epoch: [0][0/916] Time 18.342s (18.342s) Speed 0.9 samples/s Data 14.961s (14.961s) Loss 0.00214 (0.00214) Epoch: [0][50/916] Time 0.542s (0.880s) Speed 29.5 samples/s Data 0.000s (0.294s) Loss 0.00076 (0.00085) Epoch: [0][100/916] Time 0.537s (0.708s) Speed 29.8 samples/s Data 0.000s (0.148s) Loss 0.00074 (0.00080) Epoch: [0][150/916] Time 0.530s (0.650s) Speed 30.2 samples/s Data 0.000s (0.099s) Loss 0.00075 (0.00079) Epoch: [0][200/916] Time 0.531s (0.621s) Speed 30.1 samples/s Data 0.001s (0.075s) Loss 0.00074 (0.00077) Epoch: [0][250/916] Time 0.532s (0.603s) Speed 30.1 samples/s Data 0.000s (0.060s) Loss 0.00072 (0.00077) Epoch: [0][300/916] Time 0.525s (0.592s) Speed 30.5 samples/s Data 0.000s (0.050s) Loss 0.00073 (0.00076) Epoch: [0][350/916] Time 0.541s (0.583s) Speed 29.6 samples/s Data 0.000s (0.043s) Loss 0.00071 (0.00075) Epoch: [0][400/916] Time 0.536s (0.577s) Speed 29.9 samples/s Data 0.000s (0.038s) Loss 0.00067 (0.00074) Epoch: [0][450/916] Time 0.534s (0.572s) Speed 30.0 samples/s Data 0.000s (0.034s) Loss 0.00057 (0.00073) Epoch: [0][500/916] Time 0.534s (0.568s) Speed 30.0 samples/s Data 0.000s (0.030s) Loss 0.00056 (0.00072) Epoch: [0][550/916] Time 0.528s (0.565s) Speed 30.3 samples/s Data 0.000s (0.027s) Loss 0.00055 (0.00071) Epoch: [0][600/916] Time 0.533s (0.562s) Speed 30.0 samples/s Data 0.001s (0.025s) Loss 0.00053 (0.00069) Epoch: [0][650/916] Time 0.528s (0.560s) Speed 30.3 samples/s Data 0.000s (0.023s) Loss 0.00051 (0.00068) Epoch: [0][700/916] Time 0.535s (0.558s) Speed 29.9 samples/s Data 0.000s (0.022s) Loss 0.00050 (0.00067) Epoch: [0][750/916] Time 0.537s (0.556s) Speed 29.8 samples/s Data 0.000s (0.020s) Loss 0.00053 (0.00066) Epoch: [0][800/916] Time 0.532s (0.555s) Speed 30.1 samples/s Data 0.000s (0.019s) Loss 0.00047 (0.00065) Epoch: [0][850/916] Time 0.531s (0.554s) Speed 30.1 samples/s Data 0.000s (0.018s) Loss 0.00051 (0.00064) Epoch: [0][900/916] Time 0.526s (0.552s) Speed 30.4 samples/s Data 0.000s (0.017s) Loss 0.00054 (0.00063) Train Epoch 0 time:0.5524 loss:0.0006 nme:0.3472 best: True Test Epoch 0 time:0.3146 loss:0.0005 nme:0.1605 [008]:0.8482 [010]:0.5162 => saving checkpoint to output\OWN\own_hrnet_w18