[tab] /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/coinrun/dynamics/causal/train_dynamics_causal_1_node_3415079.log (log)
#!/usr/bin/env bash

#SBATCH --nodes=1
#SBATCH --ntasks-per-node=4
#SBATCH --time=48:00:00
#SBATCH --partition=accelerated
#SBATCH --cpus-per-task=5
#SBATCH --gres=gpu:4
#SBATCH --output=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/coinrun/dynamics/causal/%x_%j.log
#SBATCH --error=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/coinrun/dynamics/causal/%x_%j.log
#SBATCH --job-name=train_dynamics_causal_1_node
#SBATCH --requeue
#SBATCH --signal=b:usr1@300 # 5 min before timeout

# --- signal trap to requeue job before timeout ---
requeue_job() {
    echo "[$(date)] caught sigusr1 (timeout warning), requeueing slurm job $SLURM_JOB_ID..."
    # optional: trigger checkpoint saving here
    # e.g., touch $checkpoint_dir/requeue_trigger
    scontrol requeue $SLURM_JOB_ID
    exit 0
}

trap requeue_job sigusr1

# set checkpoint flag based on restart count
restart_count=$(scontrol show job $SLURM_JOB_ID | grep -o 'Restarts=[0-9]*' | cut -d'=' -f2)

if [ $restart_count -eq 0 ]; then
    restore_ckpt_flag="--no-restore-ckpt"
else
    restore_ckpt_flag="--restore-ckpt"
fi



# Log the sbatch script
cat $0

module unload mpi/openmpi/5.0
module unload devel/cuda/12.4
source .venv/bin/activate

array_records_dir=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/data_new/open_ai_minecraft_arrayrecords_chunked

job_name=$SLURM_JOB_NAME
slurm_job_id=$SLURM_JOB_ID

CHECKPOINT_DIR=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/coinrun/causal/$job_name/$slurm_job_id
mkdir -p $CHECKPOINT_DIR

tokenizer_ckpt_dir="/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046"

env | grep SLURM

srun python train_dynamics.py \
    --save_ckpt \
    $restore_ckpt_flag \
    --wandb_id $SLURM_JOB_ID \
    --ckpt_dir $CHECKPOINT_DIR \
    --batch_size=48 \
    --dyna_type=causal \
    --init_lr=0 \
    --max_lr=1e-5 \
    --log_image_interval=1000 \
    --log \
    --log_checkpoint_interval=1000 \
    --name=coinrun-dynamics-causal-1-node-$slurm_job_id \
    --tags coinrun dynamics causal 1-node \
    --entity instant-uv \
    --project jafar \
    --tokenizer_checkpoint=$tokenizer_ckpt_dir \
    --data_dir $array_records_dir &

child_pid=$!

wait $child_pid
/var/spool/slurmd/job3415079/slurm_script: line 42: .venv/bin/activate: No such file or directory
SLURM_JOB_USER=tum_cte0515
SLURM_TASKS_PER_NODE=4
SLURM_JOB_UID=999226
SLURM_TASK_PID=2179981
SLURM_JOB_GPUS=0,1,2,3
SLURM_LOCALID=0
SLURM_SUBMIT_DIR=/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar_jobs
SLURMD_NODENAME=hkn0726
SLURM_JOB_START_TIME=1754922213
SLURM_CLUSTER_NAME=hk
SLURM_JOB_END_TIME=1755095013
SLURM_CPUS_ON_NODE=24
SLURM_JOB_CPUS_PER_NODE=24
SLURM_GPUS_ON_NODE=4
SLURM_GTIDS=0
SLURM_JOB_PARTITION=accelerated
SLURM_TRES_PER_TASK=cpu=5
SLURM_OOM_KILL_STEP=0
SLURM_JOB_NUM_NODES=1
SLURM_JOBID=3415079
SLURM_JOB_QOS=normal
SLURM_PROCID=0
SLURM_CPUS_PER_TASK=5
SLURM_NTASKS=4
SLURM_TOPOLOGY_ADDR=hkibb.hkibbi1.hkibbi1e9.hkn0726
SLURM_TOPOLOGY_ADDR_PATTERN=switch.switch.switch.node
SLURM_SCRIPT_CONTEXT=prolog_task
SLURM_NODELIST=hkn0726
SLURM_JOB_ACCOUNT=hk-project-p0023960
SLURM_PRIO_PROCESS=0
SLURM_NPROCS=4
SLURM_NNODES=1
SLURM_SUBMIT_HOST=hkn1993.localdomain
SLURM_JOB_ID=3415079
SLURM_NODEID=0
SLURM_CONF=/etc/slurm/slurm.conf
SLURM_JOB_NAME=train_dynamics_causal_1_node
SLURM_NTASKS_PER_NODE=4
SLURM_JOB_GID=502226
SLURM_JOB_NODELIST=hkn0726
GpuFreq=control_disabled
wandb: Currently logged in as: mihir-mahajan2002 (instant-uv) to https://api.wandb.ai. Use `wandb login --relogin` to force relogin
wandb: Tracking run with wandb version 0.19.11
wandb: Run data is saved locally in /hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar_jobs/wandb/run-20250811_162435-3415079
wandb: Run `wandb offline` to turn off syncing.
wandb: Syncing run coinrun-dynamics-causal-1-node-3415079
wandb: ⭐️ View project at https://wandb.ai/instant-uv/jafar
wandb: 🚀 View run at https://wandb.ai/instant-uv/jafar/runs/3415079
WARNING:absl:Dropping 2 examples of 89394 examples (shard 4).
WARNING:absl:Dropping 2 examples of 89394 examples (shard 4).
WARNING:absl:Dropping 2 examples of 89394 examples (shard 4).
WARNING:absl:Dropping 2 examples of 89394 examples (shard 4).
WARNING:absl:Missing metrics for step 40000
WARNING:absl:Missing metrics for step 40000
WARNING:absl:Missing metrics for step 40000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/040000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 40000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/040000/metrics/metrics not found.
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/040000/metrics/metrics not found.
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/040000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 56000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/056000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 56000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/056000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 56000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/056000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 56000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/056000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 57000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/057000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 57000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/057000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 57000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/057000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 57000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/057000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 20000
WARNING:absl:Missing metrics for step 20000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/020000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 20000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/020000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 20000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/020000/metrics/metrics not found.
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/020000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 58000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/058000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 58000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/058000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 58000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/058000/metrics/metrics not found.
WARNING:absl:Missing metrics for step 58000
ERROR:absl:File /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer/train_tokenizer_1e-4/3414046/058000/metrics/metrics not found.
/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1251: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1251: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1251: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1251: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
Running on 4 devices.
Counting all components: ['dynamics', 'lam', 'tokenizer']
Parameter counts:
{'dynamics': 26555392, 'lam': 35115232, 'tokenizer': 33750256, 'total': 95420880}
Starting training from step 0...
Running on 4 devices.
Counting all components: ['dynamics', 'lam', 'tokenizer']
Parameter counts:
{'dynamics': 26555392, 'lam': 35115232, 'tokenizer': 33750256, 'total': 95420880}
Starting training from step 0...
Running on 4 devices.
Counting all components: ['dynamics', 'lam', 'tokenizer']
Parameter counts:
{'dynamics': 26555392, 'lam': 35115232, 'tokenizer': 33750256, 'total': 95420880}
Starting training from step 0...
Running on 4 devices.
Counting all components: ['dynamics', 'lam', 'tokenizer']
Parameter counts:
{'dynamics': 26555392, 'lam': 35115232, 'tokenizer': 33750256, 'total': 95420880}
Starting training from step 0...
2025-08-11 16:26:15.661574: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-08-11 16:26:15.662856: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-08-11 16:26:15.662880: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-08-11 16:26:18.430125: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-08-11 16:26:18.430191: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-08-11 16:26:25.898518: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
Step 0, loss: 12.667804718017578
Step 1, loss: 12.771347999572754
Step 2, loss: 12.642227172851562
Step 3, loss: 12.731294631958008
Step 4, loss: 12.625615119934082
Step 5, loss: 12.726526260375977
Step 6, loss: 12.679959297180176
Step 7, loss: 12.740076065063477
Step 8, loss: 12.758808135986328
Step 9, loss: 12.621700286865234
Step 10, loss: 12.689375877380371
Step 11, loss: 12.711395263671875
Step 12, loss: 12.635071754455566
Step 13, loss: 12.737500190734863
Step 14, loss: 12.727999687194824
Step 15, loss: 12.6896390914917
Step 16, loss: 12.66757869720459
Step 17, loss: 12.713518142700195
Step 18, loss: 12.667643547058105
Step 19, loss: 12.741531372070312
Step 20, loss: 12.656143188476562
Step 21, loss: 12.708353996276855
Step 22, loss: 12.722424507141113
Step 23, loss: 12.613386154174805
Step 24, loss: 12.706329345703125
Step 25, loss: 12.670353889465332
Step 26, loss: 12.654339790344238
Step 27, loss: 12.607614517211914
Step 28, loss: 12.662657737731934
Step 29, loss: 12.653752326965332
Step 30, loss: 12.605743408203125
Step 31, loss: 12.71632194519043
Step 32, loss: 12.681879043579102
Step 33, loss: 12.564464569091797
Step 34, loss: 12.617006301879883
Step 35, loss: 12.60033130645752
Step 36, loss: 12.583608627319336
Step 37, loss: 12.695188522338867
Step 38, loss: 12.601780891418457
Step 39, loss: 12.555039405822754
Step 40, loss: 12.539531707763672
Step 41, loss: 12.610857009887695
Step 42, loss: 12.575074195861816
Step 43, loss: 12.597280502319336
Step 44, loss: 12.570100784301758
Step 45, loss: 12.572305679321289
Step 46, loss: 12.584064483642578
Step 47, loss: 12.492785453796387
Step 48, loss: 12.607348442077637
Step 49, loss: 12.546814918518066
Step 50, loss: 12.495633125305176
Step 51, loss: 12.483382225036621
Step 52, loss: 12.526220321655273
Step 53, loss: 12.498763084411621
Step 54, loss: 12.548930168151855
Step 55, loss: 12.573156356811523
Step 56, loss: 12.49267578125
Step 57, loss: 12.404444694519043
Step 58, loss: 12.397965431213379
Step 59, loss: 12.419803619384766
Step 60, loss: 12.445556640625
Step 61, loss: 12.421695709228516
Step 62, loss: 12.470447540283203
Step 63, loss: 12.449533462524414
Step 64, loss: 12.47229290008545
Step 65, loss: 12.380741119384766
Step 66, loss: 12.41545295715332
Step 67, loss: 12.410737991333008
Step 68, loss: 12.396747589111328
Step 69, loss: 12.404146194458008
Step 70, loss: 12.31839656829834
Step 71, loss: 12.361237525939941
Step 72, loss: 12.293877601623535
Step 73, loss: 12.361063957214355
Step 74, loss: 12.32302474975586
Step 75, loss: 12.389338493347168
Step 76, loss: 12.24377727508545
Step 77, loss: 12.283151626586914
Step 78, loss: 12.348030090332031
Step 79, loss: 12.286529541015625
Step 80, loss: 12.259624481201172
Step 81, loss: 12.337592124938965
Step 82, loss: 12.294958114624023
Step 83, loss: 12.2918062210083
Step 84, loss: 12.283763885498047
Step 85, loss: 12.357927322387695
Step 86, loss: 12.222745895385742
Step 87, loss: 12.186835289001465
Step 88, loss: 12.234064102172852
Step 89, loss: 12.225297927856445
Step 90, loss: 12.133995056152344
Step 91, loss: 12.153714179992676
Step 92, loss: 12.19133186340332
Step 93, loss: 12.185487747192383
Step 94, loss: 12.103611946105957
Step 95, loss: 12.179014205932617
Step 96, loss: 12.194933891296387
Step 97, loss: 12.12974739074707
Step 98, loss: 12.182798385620117
Step 99, loss: 12.08813190460205
Step 100, loss: 12.116350173950195
Step 101, loss: 12.13232421875
Step 102, loss: 12.077340126037598
Step 103, loss: 12.075632095336914
Step 104, loss: 12.0490140914917
Step 105, loss: 12.019495010375977
Step 106, loss: 12.09927749633789
Step 107, loss: 12.032045364379883
Step 108, loss: 12.031266212463379
Step 109, loss: 12.077627182006836
Step 110, loss: 11.903912544250488
Step 111, loss: 11.966301918029785
Step 112, loss: 11.957892417907715
Step 113, loss: 11.926132202148438
Step 114, loss: 11.861052513122559
Step 115, loss: 11.882122039794922
Step 116, loss: 11.95129108428955
Step 117, loss: 11.872302055358887
Step 118, loss: 11.946744918823242
Step 119, loss: 11.910722732543945
Step 120, loss: 11.818961143493652
Step 121, loss: 11.851935386657715
Step 122, loss: 11.751891136169434
Step 123, loss: 11.846611022949219
Step 124, loss: 11.800748825073242
Step 125, loss: 11.796424865722656
Step 126, loss: 11.81689167022705
Step 127, loss: 11.72142505645752
Step 128, loss: 11.75197982788086
Step 129, loss: 11.763428688049316
Step 130, loss: 11.691763877868652
Step 131, loss: 11.680811882019043
Step 132, loss: 11.72774887084961
Step 133, loss: 11.640332221984863
Step 134, loss: 11.707895278930664
Step 135, loss: 11.7560396194458
Step 136, loss: 11.70436954498291
Step 137, loss: 11.6412935256958
Step 138, loss: 11.607065200805664
Step 139, loss: 11.677067756652832
Step 140, loss: 11.646750450134277
Step 141, loss: 11.694757461547852
Step 142, loss: 11.600968360900879
Step 143, loss: 11.685184478759766
Step 144, loss: 11.597763061523438
Step 145, loss: 11.548248291015625
Step 146, loss: 11.546873092651367
Step 147, loss: 11.577543258666992
Step 148, loss: 11.547377586364746
Step 149, loss: 11.524612426757812
Step 150, loss: 11.507003784179688
Step 151, loss: 11.515893936157227
Step 152, loss: 11.48664379119873
Step 153, loss: 11.502408027648926
Step 154, loss: 11.445438385009766
Step 155, loss: 11.46118450164795
Step 156, loss: 11.405437469482422
Step 157, loss: 11.443511962890625
Step 158, loss: 11.451423645019531
Step 159, loss: 11.461939811706543
Step 160, loss: 11.349418640136719
Step 161, loss: 11.422706604003906
Step 162, loss: 11.401671409606934
Step 163, loss: 11.287398338317871
Step 164, loss: 11.302481651306152
Step 165, loss: 11.289762496948242
Step 166, loss: 11.334724426269531
Step 167, loss: 11.282970428466797
Step 168, loss: 11.257315635681152
Step 169, loss: 11.28621768951416
Step 170, loss: 11.20481014251709
Step 171, loss: 11.281057357788086
Step 172, loss: 11.262025833129883
Step 173, loss: 11.243955612182617
Step 174, loss: 11.169376373291016
Step 175, loss: 11.254855155944824
Step 176, loss: 11.154994010925293
Step 177, loss: 11.21339225769043
Step 178, loss: 11.251949310302734
Step 179, loss: 11.18138599395752
Step 180, loss: 11.146556854248047
Step 181, loss: 11.13279914855957
Step 182, loss: 11.167080879211426
Step 183, loss: 11.185643196105957
Step 184, loss: 11.071303367614746
Step 185, loss: 11.087559700012207
Step 186, loss: 11.119420051574707
Step 187, loss: 11.075028419494629
Step 188, loss: 11.067861557006836
Step 189, loss: 10.989562034606934
Step 190, loss: 11.052154541015625
Step 191, loss: 11.065995216369629
Step 192, loss: 11.021690368652344
Step 193, loss: 10.993510246276855
Step 194, loss: 11.0082368850708
Step 195, loss: 10.894791603088379
Step 196, loss: 11.0162935256958
Step 197, loss: 10.924046516418457
Step 198, loss: 10.87924575805664
Step 199, loss: 10.848145484924316
Step 200, loss: 10.899628639221191
Step 201, loss: 10.91595458984375
Step 202, loss: 10.843033790588379
Step 203, loss: 10.911846160888672
Step 204, loss: 10.81973648071289
Step 205, loss: 10.85107135772705
Step 206, loss: 10.819169044494629
Step 207, loss: 10.854994773864746
Step 208, loss: 10.806999206542969
Step 209, loss: 10.800849914550781
Step 210, loss: 10.799738883972168
Step 211, loss: 10.830042839050293
Step 212, loss: 10.736322402954102
Step 213, loss: 10.752872467041016
Step 214, loss: 10.725028991699219
Step 215, loss: 10.731749534606934
Step 216, loss: 10.689827919006348
Step 217, loss: 10.62132453918457
Step 218, loss: 10.703144073486328
Step 219, loss: 10.668985366821289
Step 220, loss: 10.677373886108398
Step 221, loss: 10.62880802154541
Step 222, loss: 10.65042781829834
Step 223, loss: 10.613385200500488
Step 224, loss: 10.603299140930176
Step 225, loss: 10.608231544494629
Step 226, loss: 10.598186492919922
Step 227, loss: 10.607372283935547
Step 228, loss: 10.588812828063965
Step 229, loss: 10.578028678894043
Step 230, loss: 10.530580520629883
Step 231, loss: 10.605957984924316
Step 232, loss: 10.505242347717285
Step 233, loss: 10.477911949157715
Step 234, loss: 10.557123184204102
Step 235, loss: 10.461018562316895
Step 236, loss: 10.512309074401855
Step 237, loss: 10.466352462768555
Step 0, loss: 12.667804718017578
Step 1, loss: 12.771347999572754
Step 2, loss: 12.642227172851562
Step 3, loss: 12.731294631958008
Step 4, loss: 12.625615119934082
Step 5, loss: 12.726526260375977
Step 6, loss: 12.679959297180176
Step 7, loss: 12.740076065063477
Step 8, loss: 12.758808135986328
Step 9, loss: 12.621700286865234
Step 10, loss: 12.689375877380371
Step 11, loss: 12.711395263671875
Step 12, loss: 12.635071754455566
Step 13, loss: 12.737500190734863
Step 14, loss: 12.727999687194824
Step 15, loss: 12.6896390914917
Step 16, loss: 12.66757869720459
Step 17, loss: 12.713518142700195
Step 18, loss: 12.667643547058105
Step 19, loss: 12.741531372070312
Step 20, loss: 12.656143188476562
Step 21, loss: 12.708353996276855
Step 22, loss: 12.722424507141113
Step 23, loss: 12.613386154174805
Step 24, loss: 12.706329345703125
Step 25, loss: 12.670353889465332
Step 26, loss: 12.654339790344238
Step 27, loss: 12.607614517211914
Step 28, loss: 12.662657737731934
Step 29, loss: 12.653752326965332
Step 30, loss: 12.605743408203125
Step 31, loss: 12.71632194519043
Step 32, loss: 12.681879043579102
Step 33, loss: 12.564464569091797
Step 34, loss: 12.617006301879883
Step 35, loss: 12.60033130645752
Step 36, loss: 12.583608627319336
Step 37, loss: 12.695188522338867
Step 38, loss: 12.601780891418457
Step 39, loss: 12.555039405822754
Step 40, loss: 12.539531707763672
Step 41, loss: 12.610857009887695
Step 42, loss: 12.575074195861816
Step 43, loss: 12.597280502319336
Step 44, loss: 12.570100784301758
Step 45, loss: 12.572305679321289
Step 46, loss: 12.584064483642578
Step 47, loss: 12.492785453796387
Step 48, loss: 12.607348442077637
Step 49, loss: 12.546814918518066
Step 50, loss: 12.495633125305176
Step 51, loss: 12.483382225036621
Step 52, loss: 12.526220321655273
Step 53, loss: 12.498763084411621
Step 54, loss: 12.548930168151855
Step 55, loss: 12.573156356811523
Step 56, loss: 12.49267578125
Step 57, loss: 12.404444694519043
Step 58, loss: 12.397965431213379
Step 59, loss: 12.419803619384766
Step 60, loss: 12.445556640625
Step 61, loss: 12.421695709228516
Step 62, loss: 12.470447540283203
Step 63, loss: 12.449533462524414
Step 64, loss: 12.47229290008545
Step 65, loss: 12.380741119384766
Step 66, loss: 12.41545295715332
Step 67, loss: 12.410737991333008
Step 68, loss: 12.396747589111328
Step 69, loss: 12.404146194458008
Step 70, loss: 12.31839656829834
Step 71, loss: 12.361237525939941
Step 72, loss: 12.293877601623535
Step 73, loss: 12.361063957214355
Step 74, loss: 12.32302474975586
Step 75, loss: 12.389338493347168
Step 76, loss: 12.24377727508545
Step 77, loss: 12.283151626586914
Step 78, loss: 12.348030090332031
Step 79, loss: 12.286529541015625
Step 80, loss: 12.259624481201172
Step 81, loss: 12.337592124938965
Step 82, loss: 12.294958114624023
Step 83, loss: 12.2918062210083
Step 84, loss: 12.283763885498047
Step 85, loss: 12.357927322387695
Step 86, loss: 12.222745895385742
Step 87, loss: 12.186835289001465
Step 88, loss: 12.234064102172852
Step 89, loss: 12.225297927856445
Step 90, loss: 12.133995056152344
Step 91, loss: 12.153714179992676
Step 92, loss: 12.19133186340332
Step 93, loss: 12.185487747192383
Step 94, loss: 12.103611946105957
Step 95, loss: 12.179014205932617
Step 96, loss: 12.194933891296387
Step 97, loss: 12.12974739074707
Step 98, loss: 12.182798385620117
Step 99, loss: 12.08813190460205
Step 100, loss: 12.116350173950195
Step 101, loss: 12.13232421875
Step 102, loss: 12.077340126037598
Step 103, loss: 12.075632095336914
Step 104, loss: 12.0490140914917
Step 105, loss: 12.019495010375977
Step 106, loss: 12.09927749633789
Step 107, loss: 12.032045364379883
Step 108, loss: 12.031266212463379
Step 109, loss: 12.077627182006836
Step 110, loss: 11.903912544250488
Step 111, loss: 11.966301918029785
Step 112, loss: 11.957892417907715
Step 113, loss: 11.926132202148438
Step 114, loss: 11.861052513122559
Step 115, loss: 11.882122039794922
Step 116, loss: 11.95129108428955
Step 117, loss: 11.872302055358887
Step 118, loss: 11.946744918823242
Step 119, loss: 11.910722732543945
Step 120, loss: 11.818961143493652
Step 0, loss: 12.667804718017578
Step 1, loss: 12.771347999572754
Step 2, loss: 12.642227172851562
Step 3, loss: 12.731294631958008
Step 4, loss: 12.625615119934082
Step 5, loss: 12.726526260375977
Step 6, loss: 12.679959297180176
Step 7, loss: 12.740076065063477
Step 8, loss: 12.758808135986328
Step 9, loss: 12.621700286865234
Step 10, loss: 12.689375877380371
Step 11, loss: 12.711395263671875
Step 12, loss: 12.635071754455566
Step 13, loss: 12.737500190734863
Step 14, loss: 12.727999687194824
Step 15, loss: 12.6896390914917
Step 16, loss: 12.66757869720459
Step 17, loss: 12.713518142700195
Step 18, loss: 12.667643547058105
Step 19, loss: 12.741531372070312
Step 20, loss: 12.656143188476562
Step 21, loss: 12.708353996276855
Step 22, loss: 12.722424507141113
Step 23, loss: 12.613386154174805
Step 24, loss: 12.706329345703125
Step 25, loss: 12.670353889465332
Step 26, loss: 12.654339790344238
Step 27, loss: 12.607614517211914
Step 28, loss: 12.662657737731934
Step 29, loss: 12.653752326965332
Step 30, loss: 12.605743408203125
Step 31, loss: 12.71632194519043
Step 32, loss: 12.681879043579102
Step 33, loss: 12.564464569091797
Step 34, loss: 12.617006301879883
Step 35, loss: 12.60033130645752
Step 36, loss: 12.583608627319336
Step 37, loss: 12.695188522338867
Step 38, loss: 12.601780891418457
Step 39, loss: 12.555039405822754
Step 40, loss: 12.539531707763672
Step 41, loss: 12.610857009887695
Step 42, loss: 12.575074195861816
Step 43, loss: 12.597280502319336
Step 44, loss: 12.570100784301758
Step 45, loss: 12.572305679321289
Step 46, loss: 12.584064483642578
Step 47, loss: 12.492785453796387
Step 48, loss: 12.607348442077637
Step 49, loss: 12.546814918518066
Step 50, loss: 12.495633125305176
Step 51, loss: 12.483382225036621
Step 52, loss: 12.526220321655273
Step 53, loss: 12.498763084411621
Step 54, loss: 12.548930168151855
Step 55, loss: 12.573156356811523
Step 56, loss: 12.49267578125
Step 57, loss: 12.404444694519043
Step 58, loss: 12.397965431213379
Step 59, loss: 12.419803619384766
Step 60, loss: 12.445556640625
Step 61, loss: 12.421695709228516
Step 62, loss: 12.470447540283203
Step 63, loss: 12.449533462524414
Step 64, loss: 12.47229290008545
Step 65, loss: 12.380741119384766
Step 66, loss: 12.41545295715332
Step 67, loss: 12.410737991333008
Step 68, loss: 12.396747589111328
Step 69, loss: 12.404146194458008
Step 70, loss: 12.31839656829834
Step 71, loss: 12.361237525939941
Step 72, loss: 12.293877601623535
Step 73, loss: 12.361063957214355
Step 74, loss: 12.32302474975586
Step 75, loss: 12.389338493347168
Step 76, loss: 12.24377727508545
Step 77, loss: 12.283151626586914
Step 78, loss: 12.348030090332031
Step 79, loss: 12.286529541015625
Step 80, loss: 12.259624481201172
Step 81, loss: 12.337592124938965
Step 82, loss: 12.294958114624023
Step 83, loss: 12.2918062210083
Step 84, loss: 12.283763885498047
Step 85, loss: 12.357927322387695
Step 86, loss: 12.222745895385742
Step 87, loss: 12.186835289001465
Step 88, loss: 12.234064102172852
Step 89, loss: 12.225297927856445
Step 90, loss: 12.133995056152344
Step 91, loss: 12.153714179992676
Step 92, loss: 12.19133186340332
Step 93, loss: 12.185487747192383
Step 94, loss: 12.103611946105957
Step 95, loss: 12.179014205932617
Step 96, loss: 12.194933891296387
Step 97, loss: 12.12974739074707
Step 98, loss: 12.182798385620117
Step 99, loss: 12.08813190460205
Step 100, loss: 12.116350173950195
Step 101, loss: 12.13232421875
Step 102, loss: 12.077340126037598
Step 103, loss: 12.075632095336914
Step 104, loss: 12.0490140914917
Step 105, loss: 12.019495010375977
Step 106, loss: 12.09927749633789
Step 107, loss: 12.032045364379883
Step 108, loss: 12.031266212463379
Step 109, loss: 12.077627182006836
Step 110, loss: 11.903912544250488
Step 111, loss: 11.966301918029785
Step 112, loss: 11.957892417907715
Step 113, loss: 11.926132202148438
Step 114, loss: 11.861052513122559
Step 115, loss: 11.882122039794922
Step 116, loss: 11.95129108428955
Step 117, loss: 11.872302055358887
Step 118, loss: 11.946744918823242
Step 119, loss: 11.910722732543945
Step 120, loss: 11.818961143493652
Step 121, loss: 11.851935386657715
Step 122, loss: 11.751891136169434
Step 123, loss: 11.846611022949219
Step 124, loss: 11.800748825073242
Step 125, loss: 11.796424865722656
Step 126, loss: 11.81689167022705
Step 127, loss: 11.72142505645752
Step 128, loss: 11.75197982788086
Step 129, loss: 11.763428688049316
Step 130, loss: 11.691763877868652
Step 131, loss: 11.680811882019043
Step 132, loss: 11.72774887084961
Step 133, loss: 11.640332221984863
Step 134, loss: 11.707895278930664
Step 135, loss: 11.7560396194458
Step 136, loss: 11.70436954498291
Step 137, loss: 11.6412935256958
Step 138, loss: 11.607065200805664
Step 139, loss: 11.677067756652832
Step 140, loss: 11.646750450134277
Step 141, loss: 11.694757461547852
Step 142, loss: 11.600968360900879
Step 143, loss: 11.685184478759766
Step 144, loss: 11.597763061523438
Step 145, loss: 11.548248291015625
Step 146, loss: 11.546873092651367
Step 147, loss: 11.577543258666992
Step 148, loss: 11.547377586364746
Step 149, loss: 11.524612426757812
Step 150, loss: 11.507003784179688
Step 151, loss: 11.515893936157227
Step 152, loss: 11.48664379119873
Step 153, loss: 11.502408027648926
Step 154, loss: 11.445438385009766
Step 155, loss: 11.46118450164795
Step 156, loss: 11.405437469482422
Step 157, loss: 11.443511962890625
Step 158, loss: 11.451423645019531
Step 159, loss: 11.461939811706543
Step 160, loss: 11.349418640136719
Step 161, loss: 11.422706604003906
Step 162, loss: 11.401671409606934
Step 163, loss: 11.287398338317871
Step 164, loss: 11.302481651306152
Step 165, loss: 11.289762496948242
Step 166, loss: 11.334724426269531
Step 167, loss: 11.282970428466797
Step 168, loss: 11.257315635681152
Step 169, loss: 11.28621768951416
Step 170, loss: 11.20481014251709
Step 171, loss: 11.281057357788086
Step 172, loss: 11.262025833129883
Step 173, loss: 11.243955612182617
Step 174, loss: 11.169376373291016
Step 175, loss: 11.254855155944824
Step 176, loss: 11.154994010925293
Step 177, loss: 11.21339225769043
Step 178, loss: 11.251949310302734
Step 179, loss: 11.18138599395752
Step 180, loss: 11.146556854248047
Step 181, loss: 11.13279914855957
Step 182, loss: 11.167080879211426
Step 183, loss: 11.185643196105957
Step 184, loss: 11.071303367614746
Step 185, loss: 11.087559700012207
Step 186, loss: 11.119420051574707
Step 187, loss: 11.075028419494629
Step 188, loss: 11.067861557006836
Step 189, loss: 10.989562034606934
Step 190, loss: 11.052154541015625
Step 191, loss: 11.065995216369629
Step 192, loss: 11.021690368652344
Step 193, loss: 10.993510246276855
Step 194, loss: 11.0082368850708
Step 195, loss: 10.894791603088379
Step 196, loss: 11.0162935256958
Step 197, loss: 10.924046516418457
Step 198, loss: 10.87924575805664
Step 199, loss: 10.848145484924316
Step 200, loss: 10.899628639221191
Step 201, loss: 10.91595458984375
Step 202, loss: 10.843033790588379
Step 203, loss: 10.911846160888672
Step 204, loss: 10.81973648071289
Step 205, loss: 10.85107135772705
Step 206, loss: 10.819169044494629
Step 207, loss: 10.854994773864746
Step 208, loss: 10.806999206542969
Step 209, loss: 10.800849914550781
Step 210, loss: 10.799738883972168
Step 211, loss: 10.830042839050293
Step 212, loss: 10.736322402954102
Step 213, loss: 10.752872467041016
Step 214, loss: 10.725028991699219
Step 215, loss: 10.731749534606934
Step 216, loss: 10.689827919006348
Step 217, loss: 10.62132453918457
Step 218, loss: 10.703144073486328
Step 219, loss: 10.668985366821289
Step 220, loss: 10.677373886108398
Step 221, loss: 10.62880802154541
Step 222, loss: 10.65042781829834
Step 223, loss: 10.613385200500488
Step 224, loss: 10.603299140930176
Step 225, loss: 10.608231544494629
Step 226, loss: 10.598186492919922
Step 227, loss: 10.607372283935547
Step 228, loss: 10.588812828063965
Step 229, loss: 10.578028678894043
Step 230, loss: 10.530580520629883
Step 231, loss: 10.605957984924316
Step 232, loss: 10.505242347717285
Step 233, loss: 10.477911949157715
Step 234, loss: 10.557123184204102
Step 235, loss: 10.461018562316895
Step 236, loss: 10.512309074401855
Step 237, loss: 10.466352462768555
Step 0, loss: 12.667804718017578
Step 1, loss: 12.771347999572754
Step 2, loss: 12.642227172851562
Step 3, loss: 12.731294631958008
Step 4, loss: 12.625615119934082
Step 5, loss: 12.726526260375977
Step 6, loss: 12.679959297180176
Step 7, loss: 12.740076065063477
Step 8, loss: 12.758808135986328
Step 9, loss: 12.621700286865234
Step 10, loss: 12.689375877380371
Step 11, loss: 12.711395263671875
Step 12, loss: 12.635071754455566
Step 13, loss: 12.737500190734863
Step 14, loss: 12.727999687194824
Step 15, loss: 12.6896390914917
Step 16, loss: 12.66757869720459
Step 17, loss: 12.713518142700195
Step 18, loss: 12.667643547058105
Step 19, loss: 12.741531372070312
Step 20, loss: 12.656143188476562
Step 21, loss: 12.708353996276855
Step 22, loss: 12.722424507141113
Step 23, loss: 12.613386154174805
Step 24, loss: 12.706329345703125
Step 25, loss: 12.670353889465332
Step 26, loss: 12.654339790344238
Step 27, loss: 12.607614517211914
Step 28, loss: 12.662657737731934
Step 29, loss: 12.653752326965332
Step 30, loss: 12.605743408203125
Step 31, loss: 12.71632194519043
Step 32, loss: 12.681879043579102
Step 33, loss: 12.564464569091797
Step 34, loss: 12.617006301879883
Step 35, loss: 12.60033130645752
Step 36, loss: 12.583608627319336
Step 37, loss: 12.695188522338867
Step 38, loss: 12.601780891418457
Step 39, loss: 12.555039405822754
Step 40, loss: 12.539531707763672
Step 41, loss: 12.610857009887695
Step 42, loss: 12.575074195861816
Step 43, loss: 12.597280502319336
Step 44, loss: 12.570100784301758
Step 45, loss: 12.572305679321289
Step 46, loss: 12.584064483642578
Step 47, loss: 12.492785453796387
Step 48, loss: 12.607348442077637
Step 49, loss: 12.546814918518066
Step 50, loss: 12.495633125305176
Step 51, loss: 12.483382225036621
Step 52, loss: 12.526220321655273
Step 53, loss: 12.498763084411621
Step 54, loss: 12.548930168151855
Step 55, loss: 12.573156356811523
Step 56, loss: 12.49267578125
Step 57, loss: 12.404444694519043
Step 58, loss: 12.397965431213379
Step 59, loss: 12.419803619384766
Step 60, loss: 12.445556640625
Step 61, loss: 12.421695709228516
Step 62, loss: 12.470447540283203
Step 63, loss: 12.449533462524414
Step 64, loss: 12.47229290008545
Step 65, loss: 12.380741119384766
Step 66, loss: 12.41545295715332
Step 67, loss: 12.410737991333008
Step 68, loss: 12.396747589111328
Step 69, loss: 12.404146194458008
Step 70, loss: 12.31839656829834
Step 71, loss: 12.361237525939941
Step 72, loss: 12.293877601623535
Step 73, loss: 12.361063957214355
Step 74, loss: 12.32302474975586
Step 75, loss: 12.389338493347168
Step 76, loss: 12.24377727508545
Step 77, loss: 12.283151626586914
Step 78, loss: 12.348030090332031
Step 79, loss: 12.286529541015625
Step 80, loss: 12.259624481201172
Step 81, loss: 12.337592124938965
Step 82, loss: 12.294958114624023
Step 83, loss: 12.2918062210083
Step 84, loss: 12.283763885498047
Step 85, loss: 12.357927322387695
Step 86, loss: 12.222745895385742
Step 87, loss: 12.186835289001465
Step 88, loss: 12.234064102172852
Step 89, loss: 12.225297927856445
Step 90, loss: 12.133995056152344
Step 91, loss: 12.153714179992676
Step 92, loss: 12.19133186340332
Step 93, loss: 12.185487747192383
Step 94, loss: 12.103611946105957
Step 95, loss: 12.179014205932617
Step 96, loss: 12.194933891296387
Step 97, loss: 12.12974739074707
Step 98, loss: 12.182798385620117
Step 99, loss: 12.08813190460205
Step 100, loss: 12.116350173950195
Step 101, loss: 12.13232421875
Step 102, loss: 12.077340126037598
Step 103, loss: 12.075632095336914
Step 104, loss: 12.0490140914917
Step 105, loss: 12.019495010375977
Step 106, loss: 12.09927749633789
Step 107, loss: 12.032045364379883
Step 108, loss: 12.031266212463379
Step 109, loss: 12.077627182006836
Step 110, loss: 11.903912544250488
Step 111, loss: 11.966301918029785
Step 112, loss: 11.957892417907715
Step 113, loss: 11.926132202148438
Step 114, loss: 11.861052513122559
Step 115, loss: 11.882122039794922
Step 116, loss: 11.95129108428955
Step 117, loss: 11.872302055358887
Step 118, loss: 11.946744918823242
Step 119, loss: 11.910722732543945
Step 120, loss: 11.818961143493652
Step 121, loss: 11.851935386657715
Step 122, loss: 11.751891136169434
Step 123, loss: 11.846611022949219
Step 124, loss: 11.800748825073242
Step 125, loss: 11.796424865722656
Step 126, loss: 11.81689167022705
Step 127, loss: 11.72142505645752
Step 128, loss: 11.75197982788086
Step 129, loss: 11.763428688049316
Step 130, loss: 11.691763877868652
Step 131, loss: 11.680811882019043
Step 132, loss: 11.72774887084961
Step 133, loss: 11.640332221984863
Step 134, loss: 11.707895278930664
Step 135, loss: 11.7560396194458
Step 136, loss: 11.70436954498291
Step 137, loss: 11.6412935256958
Step 138, loss: 11.607065200805664
Step 139, loss: 11.677067756652832
Step 140, loss: 11.646750450134277
Step 141, loss: 11.694757461547852
Step 142, loss: 11.600968360900879
Step 143, loss: 11.685184478759766
Step 144, loss: 11.597763061523438
Step 145, loss: 11.548248291015625
Step 146, loss: 11.546873092651367
Step 147, loss: 11.577543258666992
Step 148, loss: 11.547377586364746
Step 149, loss: 11.524612426757812
Step 150, loss: 11.507003784179688
Step 151, loss: 11.515893936157227
Step 152, loss: 11.48664379119873
Step 153, loss: 11.502408027648926
Step 154, loss: 11.445438385009766
Step 155, loss: 11.46118450164795
Step 156, loss: 11.405437469482422
Step 157, loss: 11.443511962890625
Step 158, loss: 11.451423645019531
Step 159, loss: 11.461939811706543
Step 160, loss: 11.349418640136719
Step 161, loss: 11.422706604003906
Step 162, loss: 11.401671409606934
Step 163, loss: 11.287398338317871
Step 164, loss: 11.302481651306152
Step 165, loss: 11.289762496948242
Step 166, loss: 11.334724426269531
Step 167, loss: 11.282970428466797
Step 168, loss: 11.257315635681152
Step 169, loss: 11.28621768951416
Step 170, loss: 11.20481014251709
Step 171, loss: 11.281057357788086
Step 172, loss: 11.262025833129883
Step 173, loss: 11.243955612182617
Step 174, loss: 11.169376373291016
Step 175, loss: 11.254855155944824
Step 176, loss: 11.154994010925293
Step 177, loss: 11.21339225769043
Step 178, loss: 11.251949310302734
Step 179, loss: 11.18138599395752
Step 180, loss: 11.146556854248047
Step 181, loss: 11.13279914855957
Step 182, loss: 11.167080879211426
Step 183, loss: 11.185643196105957
Step 184, loss: 11.071303367614746
Step 185, loss: 11.087559700012207
Step 186, loss: 11.119420051574707
Step 187, loss: 11.075028419494629
Step 188, loss: 11.067861557006836
Step 189, loss: 10.989562034606934
Step 190, loss: 11.052154541015625
Step 191, loss: 11.065995216369629
Step 192, loss: 11.021690368652344
Step 193, loss: 10.993510246276855
Step 194, loss: 11.0082368850708
Step 195, loss: 10.894791603088379
Step 196, loss: 11.0162935256958
Step 197, loss: 10.924046516418457
Step 198, loss: 10.87924575805664
Step 199, loss: 10.848145484924316
Step 200, loss: 10.899628639221191
Step 201, loss: 10.91595458984375
Step 202, loss: 10.843033790588379
Step 203, loss: 10.911846160888672
Step 204, loss: 10.81973648071289
Step 205, loss: 10.85107135772705
Step 206, loss: 10.819169044494629
Step 207, loss: 10.854994773864746
Step 208, loss: 10.806999206542969
Step 209, loss: 10.800849914550781
Step 210, loss: 10.799738883972168
Step 211, loss: 10.830042839050293
Step 212, loss: 10.736322402954102
Step 213, loss: 10.752872467041016
Step 214, loss: 10.725028991699219
Step 215, loss: 10.731749534606934
Step 216, loss: 10.689827919006348
Step 217, loss: 10.62132453918457
Step 218, loss: 10.703144073486328
Step 219, loss: 10.668985366821289
Step 220, loss: 10.677373886108398
Step 221, loss: 10.62880802154541
Step 222, loss: 10.65042781829834
Step 223, loss: 10.613385200500488
Step 224, loss: 10.603299140930176
Step 225, loss: 10.608231544494629
Step 226, loss: 10.598186492919922
Step 227, loss: 10.607372283935547
Step 228, loss: 10.588812828063965
Step 229, loss: 10.578028678894043
Step 230, loss: 10.530580520629883
Step 231, loss: 10.605957984924316
Step 232, loss: 10.505242347717285
Step 233, loss: 10.477911949157715
Step 234, loss: 10.557123184204102
Step 235, loss: 10.461018562316895
Step 236, loss: 10.512309074401855
Step 237, loss: 10.466352462768555
Step 121, loss: 11.851935386657715
Step 122, loss: 11.751891136169434
Step 123, loss: 11.846611022949219
Step 124, loss: 11.800748825073242
Step 125, loss: 11.796424865722656
Step 126, loss: 11.81689167022705
Step 127, loss: 11.72142505645752
Step 128, loss: 11.75197982788086
Step 129, loss: 11.763428688049316
Step 130, loss: 11.691763877868652
Step 131, loss: 11.680811882019043
Step 132, loss: 11.72774887084961
Step 133, loss: 11.640332221984863
Step 134, loss: 11.707895278930664
Step 135, loss: 11.7560396194458
Step 136, loss: 11.70436954498291
Step 137, loss: 11.6412935256958
Step 138, loss: 11.607065200805664
Step 139, loss: 11.677067756652832
Step 140, loss: 11.646750450134277
Step 141, loss: 11.694757461547852
Step 142, loss: 11.600968360900879
Step 143, loss: 11.685184478759766
Step 144, loss: 11.597763061523438
Step 145, loss: 11.548248291015625
Step 146, loss: 11.546873092651367
Step 147, loss: 11.577543258666992
Step 148, loss: 11.547377586364746
Step 149, loss: 11.524612426757812
Step 150, loss: 11.507003784179688
Step 151, loss: 11.515893936157227
Step 152, loss: 11.48664379119873
Step 153, loss: 11.502408027648926
Step 154, loss: 11.445438385009766
Step 155, loss: 11.46118450164795
Step 156, loss: 11.405437469482422
Step 157, loss: 11.443511962890625
Step 158, loss: 11.451423645019531
Step 159, loss: 11.461939811706543
Step 160, loss: 11.349418640136719
Step 161, loss: 11.422706604003906
Step 162, loss: 11.401671409606934
Step 163, loss: 11.287398338317871
Step 164, loss: 11.302481651306152
Step 165, loss: 11.289762496948242
Step 166, loss: 11.334724426269531
Step 167, loss: 11.282970428466797
Step 168, loss: 11.257315635681152
Step 169, loss: 11.28621768951416
Step 170, loss: 11.20481014251709
Step 171, loss: 11.281057357788086
Step 172, loss: 11.262025833129883
Step 173, loss: 11.243955612182617
Step 174, loss: 11.169376373291016
Step 175, loss: 11.254855155944824
Step 176, loss: 11.154994010925293
Step 177, loss: 11.21339225769043
Step 178, loss: 11.251949310302734
Step 179, loss: 11.18138599395752
Step 180, loss: 11.146556854248047
Step 181, loss: 11.13279914855957
Step 182, loss: 11.167080879211426
Step 183, loss: 11.185643196105957
Step 184, loss: 11.071303367614746
Step 185, loss: 11.087559700012207
Step 186, loss: 11.119420051574707
Step 187, loss: 11.075028419494629
Step 188, loss: 11.067861557006836
Step 189, loss: 10.989562034606934
Step 190, loss: 11.052154541015625
Step 191, loss: 11.065995216369629
Step 192, loss: 11.021690368652344
Step 193, loss: 10.993510246276855
Step 194, loss: 11.0082368850708
Step 195, loss: 10.894791603088379
Step 196, loss: 11.0162935256958
Step 197, loss: 10.924046516418457
Step 198, loss: 10.87924575805664
Step 199, loss: 10.848145484924316
Step 200, loss: 10.899628639221191
Step 201, loss: 10.91595458984375
Step 202, loss: 10.843033790588379
Step 203, loss: 10.911846160888672
Step 204, loss: 10.81973648071289
Step 205, loss: 10.85107135772705
Step 206, loss: 10.819169044494629
Step 207, loss: 10.854994773864746
Step 208, loss: 10.806999206542969
Step 209, loss: 10.800849914550781
Step 210, loss: 10.799738883972168
Step 211, loss: 10.830042839050293
Step 212, loss: 10.736322402954102
Step 213, loss: 10.752872467041016
Step 214, loss: 10.725028991699219
Step 215, loss: 10.731749534606934
Step 216, loss: 10.689827919006348
Step 217, loss: 10.62132453918457
Step 218, loss: 10.703144073486328
Step 219, loss: 10.668985366821289
Step 220, loss: 10.677373886108398
Step 221, loss: 10.62880802154541
Step 222, loss: 10.65042781829834
Step 223, loss: 10.613385200500488
Step 224, loss: 10.603299140930176
Step 225, loss: 10.608231544494629
Step 226, loss: 10.598186492919922
Step 227, loss: 10.607372283935547
Step 228, loss: 10.588812828063965
Step 229, loss: 10.578028678894043
Step 230, loss: 10.530580520629883
Step 231, loss: 10.605957984924316
Step 232, loss: 10.505242347717285
Step 233, loss: 10.477911949157715
Step 234, loss: 10.557123184204102
Step 235, loss: 10.461018562316895
Step 236, loss: 10.512309074401855
Step 237, loss: 10.466352462768555