[tab] /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_alfred/logs_training_tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474.log (log)
#!/usr/bin/env bash

#SBATCH --nodes=8
#SBATCH --ntasks-per-node=4
#SBATCH --time=10:00:00
#SBATCH --account=hk-project-p0023960
#SBATCH --partition=accelerated
#SBATCH --cpus-per-task=8
#SBATCH --gres=gpu:4
#SBATCH --output=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_alfred/logs_training_tokenizer/%x_%j.log
#SBATCH --error=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_alfred/logs_training_tokenizer/%x_%j.log
#SBATCH --mail-user=avocadoaling@gmail.com
#SBATCH --mail-type=ALL
#SBATCH --job-name=train_tokenizer_batch_size_scaling_8_node

# Log the sbatch script
cat $0

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

JOB_NAME=$SLURM_JOB_NAME
SLURM_JOB_ID=$SLURM_JOB_ID
TAGS=(tokenizer batch-size-scaling 8-node time-step adjusted-lr)

WS_DIR='/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/'

TF_RECORDS_DIR=$WS_DIR/data/open_ai_minecraft_tfrecord
CHECKPOINT_DIR=$WS_DIR/checkpoints/tokenizer/${JOB_NAME}_${SLURM_JOB_ID}
mkdir -p $CHECKPOINT_DIR

env | grep SLURM

srun python train_tokenizer.py \
    --ckpt_dir $CHECKPOINT_DIR \
    --batch_size=384 \
    --min_lr=8.49e-4 \
    --max_lr=8.49e-4 \
    --log_image_interval=250 \
    --log \
    --name=${JOB_NAME}-${SLURM_JOB_ID} \
    --tags $TAGS \
    --entity instant-uv \
    --project jafar \
    --log_checkpoint_interval=1000 \
    --data_dir $TF_RECORDS_DIR
SLURM_JOB_USER=tum_ind3695
SLURM_TASKS_PER_NODE=4(x8)
SLURM_JOB_UID=991285
SLURM_TASK_PID=634526
SLURM_JOB_GPUS=0,1,2,3
SLURM_LOCALID=0
SLURM_SUBMIT_DIR=/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling
SLURMD_NODENAME=hkn0420
SLURM_JOB_START_TIME=1751318063
SLURM_CLUSTER_NAME=hk
SLURM_JOB_END_TIME=1751354063
SLURM_CPUS_ON_NODE=32
SLURM_JOB_CPUS_PER_NODE=32(x8)
SLURM_GPUS_ON_NODE=4
SLURM_GTIDS=0
SLURM_JOB_PARTITION=accelerated
SLURM_TRES_PER_TASK=cpu=8
SLURM_OOM_KILL_STEP=0
SLURM_JOB_NUM_NODES=8
SLURM_JOBID=3307474
SLURM_JOB_QOS=normal
SLURM_PROCID=0
SLURM_CPUS_PER_TASK=8
SLURM_NTASKS=32
SLURM_TOPOLOGY_ADDR=hkibb.hkibbi1.hkibbi1e12.hkn0420
SLURM_TOPOLOGY_ADDR_PATTERN=switch.switch.switch.node
SLURM_SCRIPT_CONTEXT=prolog_task
SLURM_NODELIST=hkn[0420,0516,0520,0523,0704,0717,0810,0820]
SLURM_JOB_ACCOUNT=hk-project-p0023960
SLURM_PRIO_PROCESS=0
SLURM_NPROCS=32
SLURM_NNODES=8
SLURM_SUBMIT_HOST=hkn1991.localdomain
SLURM_JOB_ID=3307474
SLURM_NODEID=0
SLURM_CONF=/etc/slurm/slurm.conf
SLURM_JOB_NAME=train_tokenizer_batch_size_scaling_8_node
SLURM_NTASKS_PER_NODE=4
SLURM_JOB_GID=502289
SLURM_JOB_NODELIST=hkn[0420,0516,0520,0523,0704,0717,0810,0820]
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
2025-06-30 23:14:53.456775: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:53.456867: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:53.457066: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:53.457602: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318093.469657 2856352 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318093.469881 2856354 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318093.470016 2856351 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318093.470869 2856353 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318093.474025 2856352 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318093.474537 2856354 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318093.474636 2856351 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318093.475228 2856353 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
W0000 00:00:1751318093.487901 2856353 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.487916 2856353 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.487918 2856353 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.487919 2856353 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488292 2856352 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488308 2856352 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488310 2856352 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488312 2856352 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488485 2856351 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488502 2856351 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488503 2856351 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488505 2856351 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488598 2856354 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488615 2856354 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488617 2856354 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318093.488619 2856354 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-06-30 23:14:54.547413: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.547731: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.547735: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.547955: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.560520 2160604 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.560830 2160602 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.560803 2160603 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.561295 2160605 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318094.565025 2160604 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.565476 2160602 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.565428 2160603 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.566045 2160605 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
2025-06-30 23:14:54.566659: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.566657: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.566878: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.567170: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
W0000 00:00:1751318094.578832 2160604 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.578847 2160604 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.578849 2160604 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.578850 2160604 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579318 2160602 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579333 2160602 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579335 2160602 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579337 2160602 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579319 2160603 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579333 2160603 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579336 2160603 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579337 2160603 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579563 2160605 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579580 2160605 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579582 2160605 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.579583 2160605 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.579697 2427863 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.579746 2427864 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.579766 2427865 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.580755 2427862 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318094.584398 2427863 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.584385 2427864 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.584400 2427865 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.585575 2427862 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
W0000 00:00:1751318094.598710 2427863 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598724 2427863 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598725 2427863 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598727 2427863 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598695 2427864 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598711 2427864 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598713 2427864 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598715 2427864 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598698 2427865 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598713 2427865 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598716 2427865 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.598717 2427865 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.599187 2427862 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.599205 2427862 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.599207 2427862 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.599208 2427862 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-06-30 23:14:54.661486: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.661507: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.661685: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:14:54.661718: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.674450  634594 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.674449  634595 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.674517  634596 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318094.674538  634593 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318094.678694  634596 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.678813  634594 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.679115  634593 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318094.679028  634595 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
W0000 00:00:1751318094.692678  634593 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692692  634593 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692694  634593 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692696  634593 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692652  634594 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692665  634594 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692667  634594 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692668  634594 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692666  634595 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692682  634595 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692683  634595 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692685  634595 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692627  634596 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692643  634596 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692645  634596 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318094.692646  634596 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318099.464209 2856354 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.464573 2856353 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.464844 2856351 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.467389 2856352 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.527601 2160602 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.527693 2160603 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.527574 2160604 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.527854 2160605 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.612797 2427862 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.612805 2427863 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.612785 2427864 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318099.612795 2427865 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318100.065098  634594 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318100.065192  634595 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318100.065238  634593 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318100.065260  634596 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
2025-06-30 23:15:01.213957: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:01.213951: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:01.213959: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:01.213955: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318101.384363 1578014 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318101.384257 1578015 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318101.384268 1578017 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318101.384491 1578016 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318101.391716 1578014 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318101.391711 1578015 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318101.391718 1578016 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318101.391711 1578017 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
W0000 00:00:1751318101.879162 1578014 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879190 1578014 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879193 1578014 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879194 1578014 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879159 1578015 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879189 1578015 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879190 1578015 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879192 1578015 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879165 1578016 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879191 1578016 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879193 1578016 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879195 1578016 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879160 1578017 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879185 1578017 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879187 1578017 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318101.879188 1578017 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-06-30 23:15:02.184361: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.184363: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.184357: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.184357: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.214725 1012654 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.214631 1012655 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.214813 1012656 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.214638 1012657 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318102.234122 1012655 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.234118 1012657 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.234789 1012654 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.234772 1012656 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
W0000 00:00:1751318102.361690 1012654 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361715 1012654 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361717 1012654 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361719 1012654 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361694 1012655 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361718 1012655 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361720 1012655 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361721 1012655 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361687 1012656 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361712 1012656 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361715 1012656 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361716 1012656 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361696 1012657 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361714 1012657 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361717 1012657 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.361718 1012657 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-06-30 23:15:02.368657: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.368656: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.368655: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.368654: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.398233 3005151 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.398299 3005153 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.398548 3005152 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.398657 3005154 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318102.404511 3005151 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.404506 3005152 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.404509 3005153 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.404508 3005154 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
2025-06-30 23:15:02.421302: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.421417: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.421303: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-06-30 23:15:02.421425: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:467] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.455392 3894030 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.455330 3894031 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.455513 3894032 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1751318102.455482 3894033 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1751318102.461184 3894031 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.461184 3894032 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.461181 3894033 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
E0000 00:00:1751318102.462009 3894030 cuda_blas.cc:1407] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
W0000 00:00:1751318102.504381 3005151 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504408 3005151 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504410 3005151 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504412 3005151 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504377 3005152 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504402 3005152 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504405 3005152 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504407 3005152 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504382 3005153 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504409 3005153 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504412 3005153 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504414 3005153 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504380 3005154 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504405 3005154 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504407 3005154 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.504409 3005154 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534858 3894030 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534885 3894030 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534887 3894030 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534889 3894030 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534860 3894031 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534886 3894031 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534888 3894031 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534890 3894031 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534857 3894032 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534883 3894032 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534885 3894032 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534887 3894032 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534855 3894033 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534878 3894033 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534881 3894033 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318102.534883 3894033 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751318118.615248 1578017 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318118.615450 1578015 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318118.615611 1578014 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318118.615637 1578016 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318120.944360 3894031 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318120.944433 3894032 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318120.944359 3894033 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318120.944517 3894030 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318122.638730 3005151 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318122.638731 3005153 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318122.638863 3005154 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318122.638879 3005152 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318123.581820 1012654 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318123.581743 1012655 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318123.581753 1012657 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
W0000 00:00:1751318123.581744 1012656 gpu_device.cc:2341] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
wandb: Currently logged in as: avocadoali (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-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/wandb/run-20250630_231525-hzw9mfvm
wandb: Run `wandb offline` to turn off syncing.
wandb: Syncing run train_tokenizer_batch_size_scaling_8_node-3307474
wandb: ⭐️ View project at https://wandb.ai/instant-uv/jafar
wandb: 🚀 View run at https://wandb.ai/instant-uv/jafar/runs/hzw9mfvm
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 115, loss: 0.17339496314525604
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 115, loss: 0.17339496314525604
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 115, loss: 0.17339496314525604
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 115, loss: 0.17339496314525604
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 115, loss: 0.17339496314525604
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 115, loss: 0.17339496314525604
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 115, loss: 0.17339496314525604
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 115, loss: 0.17339496314525604
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 115, loss: 0.17339496314525604
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 115, loss: 0.17339496314525604
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 115, loss: 0.17339496314525604
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 115, loss: 0.17339496314525604
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 115, loss: 0.17339496314525604
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Running on 32 devices.
Starting training from step 0...
Step 0, loss: 0.3471579849720001
Step 1, loss: 0.26770803332328796
Step 2, loss: 0.14104889333248138
Step 3, loss: 0.13309144973754883
Step 4, loss: 0.13091082870960236
Step 5, loss: 0.13509593904018402
Step 6, loss: 0.15321968495845795
Step 7, loss: 0.14519496262073517
Step 8, loss: 0.14099560678005219
Step 9, loss: 0.1463150680065155
Step 10, loss: 0.14669667184352875
Step 11, loss: 0.1430281549692154
Step 12, loss: 0.13978496193885803
Step 13, loss: 0.13720174133777618
Step 14, loss: 0.14147496223449707
Step 15, loss: 0.12998127937316895
Step 16, loss: 0.12920445203781128
Step 17, loss: 0.13227719068527222
Step 18, loss: 0.125120609998703
Step 19, loss: 0.13470731675624847
Step 20, loss: 0.12474726885557175
Step 21, loss: 0.1333005726337433
Step 22, loss: 0.1423025131225586
Step 23, loss: 0.1377020627260208
Step 24, loss: 0.1437397450208664
Step 25, loss: 0.13958904147148132
Step 26, loss: 0.13821439445018768
Step 27, loss: 0.14022579789161682
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 115, loss: 0.17339496314525604
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 115, loss: 0.17339496314525604
Step 115, loss: 0.17339496314525604
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 28, loss: 0.13478153944015503
Step 29, loss: 0.1434846669435501
Step 30, loss: 0.13330970704555511
Step 31, loss: 0.14295107126235962
Step 32, loss: 0.14157991111278534
Step 33, loss: 0.13778111338615417
Step 34, loss: 0.1367509663105011
Step 35, loss: 0.14291462302207947
Step 36, loss: 0.13713470101356506
Step 37, loss: 0.1495264619588852
Step 38, loss: 0.15450292825698853
Step 39, loss: 0.1477455347776413
Step 40, loss: 0.14942966401576996
Step 41, loss: 0.15012970566749573
Step 42, loss: 0.14514152705669403
Step 43, loss: 0.1656753122806549
Step 44, loss: 0.15873289108276367
Step 45, loss: 0.17154277861118317
Step 46, loss: 0.17314167320728302
Step 47, loss: 0.17424646019935608
Step 48, loss: 0.17465342581272125
Step 49, loss: 0.17834438383579254
Step 50, loss: 0.1750008463859558
Step 51, loss: 0.17383235692977905
Step 52, loss: 0.1738968789577484
Step 53, loss: 0.1788439154624939
Step 54, loss: 0.17455320060253143
Step 55, loss: 0.17899788916110992
Step 56, loss: 0.17514869570732117
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 57, loss: 0.17526665329933167
Step 58, loss: 0.17442293465137482
Step 59, loss: 0.1755731999874115
Step 60, loss: 0.17686791718006134
Step 61, loss: 0.1774514615535736
Step 62, loss: 0.17734190821647644
Step 63, loss: 0.17445750534534454
Step 64, loss: 0.1791164129972458
Step 65, loss: 0.17443639039993286
Step 66, loss: 0.1721915602684021
Step 67, loss: 0.17724011838436127
Step 68, loss: 0.1726222038269043
Step 69, loss: 0.17514818906784058
Step 70, loss: 0.17387129366397858
Step 71, loss: 0.17663298547267914
Step 72, loss: 0.17917156219482422
Step 73, loss: 0.17480577528476715
Step 74, loss: 0.17429769039154053
Step 75, loss: 0.17326530814170837
Step 76, loss: 0.17588405311107635
Step 77, loss: 0.16884107887744904
Step 78, loss: 0.17379479110240936
Step 79, loss: 0.17513428628444672
Step 80, loss: 0.17434917390346527
Step 81, loss: 0.17318695783615112
Step 82, loss: 0.17380864918231964
Step 83, loss: 0.17578117549419403
Step 84, loss: 0.1710706204175949
Step 85, loss: 0.17886455357074738
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 86, loss: 0.17388099431991577
Step 87, loss: 0.17131978273391724
Step 88, loss: 0.17590537667274475
Step 89, loss: 0.1696556806564331
Step 90, loss: 0.1746663749217987
Step 91, loss: 0.1718718260526657
Step 92, loss: 0.17620952427387238
Step 93, loss: 0.17497602105140686
Step 94, loss: 0.17112667858600616
Step 95, loss: 0.17390812933444977
Step 96, loss: 0.17191694676876068
Step 97, loss: 0.17629854381084442
Step 98, loss: 0.17539462447166443
Step 99, loss: 0.173387348651886
Step 100, loss: 0.17558738589286804
Step 101, loss: 0.17344261705875397
Step 102, loss: 0.17063868045806885
Step 103, loss: 0.17445780336856842
Step 104, loss: 0.17477689683437347
Step 105, loss: 0.1743089109659195
Step 106, loss: 0.17661947011947632
Step 107, loss: 0.17260797321796417
Step 108, loss: 0.17181360721588135
Step 109, loss: 0.17290376126766205
Step 110, loss: 0.17456933856010437
Step 111, loss: 0.17321565747261047
Step 112, loss: 0.17227257788181305
Step 113, loss: 0.1743168830871582
Step 114, loss: 0.16588141024112701
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 115, loss: 0.17339496314525604
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 116, loss: 0.17650632560253143
Step 117, loss: 0.1721123307943344
Step 118, loss: 0.17484426498413086
Step 119, loss: 0.17425554990768433
Step 120, loss: 0.17355427145957947
Step 121, loss: 0.17475546896457672
Step 122, loss: 0.1722601056098938
Step 123, loss: 0.17617866396903992
Step 124, loss: 0.1784822940826416
Step 125, loss: 0.17228849232196808
Step 126, loss: 0.1726805716753006
Step 127, loss: 0.17028960585594177
Step 128, loss: 0.1759348213672638
Step 129, loss: 0.1740720421075821
Step 130, loss: 0.17981602251529694
Step 131, loss: 0.1703324317932129
Step 132, loss: 0.17444254457950592
Step 133, loss: 0.17495402693748474
Step 134, loss: 0.1774214208126068
Step 135, loss: 0.16589730978012085
Step 136, loss: 0.17302198708057404
Step 137, loss: 0.17555880546569824
Step 138, loss: 0.1742158681154251
Step 139, loss: 0.172081857919693
Step 140, loss: 0.17138473689556122
Step 141, loss: 0.17552287876605988
Step 142, loss: 0.17444556951522827
Step 143, loss: 0.1737712323665619
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 144, loss: 0.17160776257514954
Step 145, loss: 0.17339970171451569
Step 146, loss: 0.17081758379936218
Step 147, loss: 0.16918820142745972
Step 148, loss: 0.1746884435415268
Step 149, loss: 0.17227376997470856
Step 150, loss: 0.17421938478946686
Step 151, loss: 0.1723541021347046
Step 152, loss: 0.17495852708816528
Step 153, loss: 0.1736605316400528
Step 154, loss: 0.1731863170862198
Step 155, loss: 0.173353374004364
Step 156, loss: 0.17461292445659637
Step 157, loss: 0.17833106219768524
Step 158, loss: 0.1696862280368805
Step 159, loss: 0.17354346811771393
Step 160, loss: 0.17869757115840912
Step 161, loss: 0.16943944990634918
Step 162, loss: 0.17502373456954956
Step 163, loss: 0.17490755021572113
Step 164, loss: 0.1724388152360916
Step 165, loss: 0.1717279553413391
Step 166, loss: 0.1696639358997345
Step 167, loss: 0.1763082593679428
Step 168, loss: 0.17012852430343628
Step 169, loss: 0.1756332814693451
Step 170, loss: 0.17349863052368164
Step 171, loss: 0.17180608212947845
Step 172, loss: 0.1735956072807312
Step 173, loss: 0.16950492560863495
Step 174, loss: 0.1710001528263092
Step 175, loss: 0.1720818281173706
Step 176, loss: 0.17734099924564362
Step 177, loss: 0.17544645071029663
Step 178, loss: 0.17164328694343567
Step 179, loss: 0.17365553975105286
Step 180, loss: 0.17731669545173645
Step 181, loss: 0.16690389811992645
Step 182, loss: 0.1742502599954605
Step 183, loss: 0.1716139167547226
Step 184, loss: 0.1698140799999237
Step 185, loss: 0.17759019136428833
Step 186, loss: 0.17442889511585236
Step 187, loss: 0.17123228311538696
Step 188, loss: 0.17149075865745544
Step 189, loss: 0.17389832437038422
Step 190, loss: 0.1708046942949295
Step 191, loss: 0.171993225812912
Step 192, loss: 0.17431600391864777
Step 193, loss: 0.17453940212726593
Step 194, loss: 0.1709374040365219
Step 195, loss: 0.17524591088294983
Step 196, loss: 0.17337879538536072
Step 197, loss: 0.1743336170911789
Step 198, loss: 0.1775231659412384
Step 199, loss: 0.16916300356388092
Step 200, loss: 0.17302356660366058
Step 201, loss: 0.17107906937599182
Step 202, loss: 0.17300015687942505
Step 203, loss: 0.1730821281671524
Step 204, loss: 0.17554880678653717
Step 205, loss: 0.1745760142803192
Step 206, loss: 0.17443722486495972
Step 207, loss: 0.17716984450817108
Step 208, loss: 0.178369402885437
Step 209, loss: 0.17524774372577667
Step 210, loss: 0.17759115993976593
Step 211, loss: 0.17509549856185913
Step 212, loss: 0.1736840456724167
Step 213, loss: 0.1769413948059082
Step 214, loss: 0.17503388226032257
Step 215, loss: 0.17699705064296722
Step 216, loss: 0.17280948162078857
Step 217, loss: 0.17379453778266907
Step 218, loss: 0.17551490664482117
Step 219, loss: 0.1724730134010315
Step 220, loss: 0.17256437242031097
Step 221, loss: 0.17449802160263062
Step 222, loss: 0.1751682460308075
Step 223, loss: 0.17546094954013824
Step 224, loss: 0.17379115521907806
Step 225, loss: 0.1778661012649536
Step 226, loss: 0.17416249215602875
Step 227, loss: 0.17225857079029083
Step 228, loss: 0.17188940942287445
Step 229, loss: 0.17189325392246246
Step 230, loss: 0.17526964843273163
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 459, loss: 0.17431780695915222
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 459, loss: 0.17431780695915222
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 459, loss: 0.17431780695915222
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 459, loss: 0.17431780695915222
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 459, loss: 0.17431780695915222
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 459, loss: 0.17431780695915222
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 459, loss: 0.17431780695915222
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 459, loss: 0.17431780695915222
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 459, loss: 0.17431780695915222
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 459, loss: 0.17431780695915222
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 459, loss: 0.17431780695915222
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 459, loss: 0.17431780695915222
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 459, loss: 0.17431780695915222
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 231, loss: 0.17420436441898346
Step 232, loss: 0.1751464307308197
Step 233, loss: 0.17173068225383759
Step 234, loss: 0.17404291033744812
Step 235, loss: 0.174527108669281
Step 236, loss: 0.17618782818317413
Step 237, loss: 0.17089204490184784
Step 238, loss: 0.17432117462158203
Step 239, loss: 0.17289616167545319
Step 240, loss: 0.17036113142967224
Step 241, loss: 0.17384111881256104
Step 242, loss: 0.17676493525505066
Step 243, loss: 0.17606355249881744
Step 244, loss: 0.17296987771987915
Step 245, loss: 0.17211146652698517
Step 246, loss: 0.16899186372756958
Step 247, loss: 0.1751350313425064
Step 248, loss: 0.17139440774917603
Step 249, loss: 0.175702303647995
Step 250, loss: 0.17180339992046356
Step 251, loss: 0.1714021861553192
Step 252, loss: 0.17811693251132965
Step 253, loss: 0.1751716434955597
Step 254, loss: 0.17288176715373993
Step 255, loss: 0.1767214834690094
Step 256, loss: 0.17531660199165344
Step 257, loss: 0.1743871569633484
Step 258, loss: 0.17458227276802063
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 259, loss: 0.1700548678636551
Step 260, loss: 0.17513228952884674
Step 261, loss: 0.16957631707191467
Step 262, loss: 0.17602184414863586
Step 263, loss: 0.16820436716079712
Step 264, loss: 0.17445197701454163
Step 265, loss: 0.17325876653194427
Step 266, loss: 0.17151372134685516
Step 267, loss: 0.17281821370124817
Step 268, loss: 0.17472431063652039
Step 269, loss: 0.17343536019325256
Step 270, loss: 0.174852654337883
Step 271, loss: 0.17498376965522766
Step 272, loss: 0.17547272145748138
Step 273, loss: 0.17591245472431183
Step 274, loss: 0.1728111356496811
Step 275, loss: 0.17593519389629364
Step 276, loss: 0.17476886510849
Step 277, loss: 0.17651814222335815
Step 278, loss: 0.17160353064537048
Step 279, loss: 0.16974833607673645
Step 280, loss: 0.1757923811674118
Step 281, loss: 0.1767502725124359
Step 282, loss: 0.16855189204216003
Step 283, loss: 0.1713992804288864
Step 284, loss: 0.17589080333709717
Step 285, loss: 0.17468930780887604
Step 286, loss: 0.17121709883213043
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 287, loss: 0.17423152923583984
Step 288, loss: 0.17463481426239014
Step 289, loss: 0.17352049052715302
Step 290, loss: 0.17575746774673462
Step 291, loss: 0.17079386115074158
Step 292, loss: 0.1753157377243042
Step 293, loss: 0.17148235440254211
Step 294, loss: 0.17589227855205536
Step 295, loss: 0.17230594158172607
Step 296, loss: 0.17561212182044983
Step 297, loss: 0.17357952892780304
Step 298, loss: 0.17742855846881866
Step 299, loss: 0.17149633169174194
Step 300, loss: 0.17210690677165985
Step 301, loss: 0.17151214182376862
Step 302, loss: 0.17388419806957245
Step 303, loss: 0.1740708202123642
Step 304, loss: 0.17560219764709473
Step 305, loss: 0.17067687213420868
Step 306, loss: 0.16773159801959991
Step 307, loss: 0.171116903424263
Step 308, loss: 0.1745900809764862
Step 309, loss: 0.17308343946933746
Step 310, loss: 0.17476658523082733
Step 311, loss: 0.17111080884933472
Step 312, loss: 0.17190101742744446
Step 313, loss: 0.16911008954048157
Step 314, loss: 0.1785719096660614
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 315, loss: 0.17609170079231262
Step 316, loss: 0.1770344078540802
Step 317, loss: 0.17321272194385529
Step 318, loss: 0.1736145317554474
Step 319, loss: 0.17849916219711304
Step 320, loss: 0.16995446383953094
Step 321, loss: 0.17075978219509125
Step 322, loss: 0.1684204787015915
Step 323, loss: 0.1707623153924942
Step 324, loss: 0.16800695657730103
Step 325, loss: 0.1711212694644928
Step 326, loss: 0.17421497404575348
Step 327, loss: 0.17173270881175995
Step 328, loss: 0.1733817309141159
Step 329, loss: 0.1723349690437317
Step 330, loss: 0.17115896940231323
Step 331, loss: 0.17340612411499023
Step 332, loss: 0.17287622392177582
Step 333, loss: 0.17327342927455902
Step 334, loss: 0.17215271294116974
Step 335, loss: 0.17460770905017853
Step 336, loss: 0.17455795407295227
Step 337, loss: 0.17333874106407166
Step 338, loss: 0.1695828139781952
Step 339, loss: 0.17012254893779755
Step 340, loss: 0.17383894324302673
Step 341, loss: 0.1735590249300003
Step 342, loss: 0.17276807129383087
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 343, loss: 0.16952309012413025
Step 344, loss: 0.17104601860046387
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 345, loss: 0.17573249340057373
Step 346, loss: 0.1728946715593338
Step 347, loss: 0.17545942962169647
Step 348, loss: 0.17130595445632935
Step 349, loss: 0.17322826385498047
Step 350, loss: 0.1729235053062439
Step 351, loss: 0.17295922338962555
Step 352, loss: 0.1735742837190628
Step 353, loss: 0.1697748452425003
Step 354, loss: 0.17621870338916779
Step 355, loss: 0.1742292046546936
Step 356, loss: 0.17279407382011414
Step 357, loss: 0.17330560088157654
Step 358, loss: 0.17479054629802704
Step 359, loss: 0.17317044734954834
Step 360, loss: 0.17544831335544586
Step 361, loss: 0.17814604938030243
Step 362, loss: 0.17601491510868073
Step 363, loss: 0.17304247617721558
Step 364, loss: 0.17508576810359955
Step 365, loss: 0.1717367172241211
Step 366, loss: 0.1739933043718338
Step 367, loss: 0.17015735805034637
Step 368, loss: 0.16894015669822693
Step 369, loss: 0.17233464121818542
Step 370, loss: 0.17584733664989471
Step 371, loss: 0.17348554730415344
Step 372, loss: 0.17454135417938232
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 373, loss: 0.1766464114189148
Step 374, loss: 0.17427922785282135
Step 375, loss: 0.16924604773521423
Step 376, loss: 0.1737813949584961
Step 377, loss: 0.17719173431396484
Step 378, loss: 0.17514315247535706
Step 379, loss: 0.173935204744339
Step 380, loss: 0.17518827319145203
Step 381, loss: 0.17630116641521454
Step 382, loss: 0.17496182024478912
Step 383, loss: 0.17837847769260406
Step 384, loss: 0.1724771112203598
Step 385, loss: 0.17744266986846924
Step 386, loss: 0.17381663620471954
Step 387, loss: 0.17394840717315674
Step 388, loss: 0.17453908920288086
Step 389, loss: 0.1730494201183319
Step 390, loss: 0.17195409536361694
Step 391, loss: 0.17394614219665527
Step 392, loss: 0.17362229526042938
Step 393, loss: 0.1768149733543396
Step 394, loss: 0.16851279139518738
Step 395, loss: 0.17069624364376068
Step 396, loss: 0.17005404829978943
Step 397, loss: 0.1732345074415207
Step 398, loss: 0.17623025178909302
Step 399, loss: 0.17276795208454132
Step 400, loss: 0.1697579026222229
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 459, loss: 0.17431780695915222
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 459, loss: 0.17431780695915222
Step 401, loss: 0.1745413988828659
Step 402, loss: 0.17456986010074615
Step 403, loss: 0.17597635090351105
Step 404, loss: 0.17094317078590393
Step 405, loss: 0.17680500447750092
Step 406, loss: 0.17142018675804138
Step 407, loss: 0.1751982569694519
Step 408, loss: 0.17486493289470673
Step 409, loss: 0.17512650787830353
Step 410, loss: 0.17006459832191467
Step 411, loss: 0.16990330815315247
Step 412, loss: 0.17894293367862701
Step 413, loss: 0.1735304743051529
Step 414, loss: 0.17329593002796173
Step 415, loss: 0.17388761043548584
Step 416, loss: 0.17462217807769775
Step 417, loss: 0.17329254746437073
Step 418, loss: 0.17006121575832367
Step 419, loss: 0.17535844445228577
Step 420, loss: 0.17794457077980042
Step 421, loss: 0.17409290373325348
Step 422, loss: 0.1734360158443451
Step 423, loss: 0.17678727209568024
Step 424, loss: 0.17418088018894196
Step 425, loss: 0.17600953578948975
Step 426, loss: 0.17062707245349884
Step 427, loss: 0.17593447864055634
Step 428, loss: 0.177713543176651
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 459, loss: 0.17431780695915222
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 459, loss: 0.17431780695915222
Step 429, loss: 0.17125695943832397
Step 430, loss: 0.17908503115177155
Step 431, loss: 0.17312182486057281
Step 432, loss: 0.17226175963878632
Step 433, loss: 0.17499332129955292
Step 434, loss: 0.1692870408296585
Step 435, loss: 0.17121432721614838
Step 436, loss: 0.17658698558807373
Step 437, loss: 0.17520497739315033
Step 438, loss: 0.1722518801689148
Step 439, loss: 0.16925294697284698
Step 440, loss: 0.17579831182956696
Step 441, loss: 0.17457902431488037
Step 442, loss: 0.17101430892944336
Step 443, loss: 0.17209774255752563
Step 444, loss: 0.17333845794200897
Step 445, loss: 0.17015598714351654
Step 446, loss: 0.17298994958400726
Step 447, loss: 0.17452651262283325
Step 448, loss: 0.17778415977954865
Step 449, loss: 0.17184945940971375
Step 450, loss: 0.1754949539899826
Step 451, loss: 0.1755322515964508
Step 452, loss: 0.17120981216430664
Step 453, loss: 0.16953344643115997
Step 454, loss: 0.17265845835208893
Step 455, loss: 0.1755368858575821
Step 456, loss: 0.1723787635564804
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 457, loss: 0.17290666699409485
Step 458, loss: 0.1749171018600464
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 459, loss: 0.17431780695915222
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 688, loss: 0.168025940656662
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 688, loss: 0.168025940656662
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 688, loss: 0.168025940656662
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 688, loss: 0.168025940656662
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 688, loss: 0.168025940656662
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 688, loss: 0.168025940656662
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 688, loss: 0.168025940656662
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 688, loss: 0.168025940656662
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 460, loss: 0.1741330772638321
Step 461, loss: 0.17441166937351227
Step 462, loss: 0.1756756603717804
Step 463, loss: 0.17418205738067627
Step 464, loss: 0.17287975549697876
Step 465, loss: 0.17231808602809906
Step 466, loss: 0.17125974595546722
Step 467, loss: 0.17578507959842682
Step 468, loss: 0.1705021858215332
Step 469, loss: 0.17248119413852692
Step 470, loss: 0.17409169673919678
Step 471, loss: 0.17181046307086945
Step 472, loss: 0.17418991029262543
Step 473, loss: 0.1714387834072113
Step 474, loss: 0.17240971326828003
Step 475, loss: 0.17331163585186005
Step 476, loss: 0.17074371874332428
Step 477, loss: 0.17336028814315796
Step 478, loss: 0.17391328513622284
Step 479, loss: 0.17445111274719238
Step 480, loss: 0.17107941210269928
Step 481, loss: 0.17514654994010925
Step 482, loss: 0.17364846169948578
Step 483, loss: 0.16913525760173798
Step 484, loss: 0.17099237442016602
Step 485, loss: 0.17201489210128784
Step 486, loss: 0.17490939795970917
Step 487, loss: 0.17496316134929657
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 688, loss: 0.168025940656662
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 488, loss: 0.17247094213962555
Step 489, loss: 0.17306217551231384
Step 490, loss: 0.17290695011615753
Step 491, loss: 0.174490824341774
Step 492, loss: 0.17366597056388855
Step 493, loss: 0.17133276164531708
Step 494, loss: 0.17247363924980164
Step 495, loss: 0.17463356256484985
Step 496, loss: 0.17164246737957
Step 497, loss: 0.1705310344696045
Step 498, loss: 0.17419204115867615
Step 499, loss: 0.1729225218296051
Step 500, loss: 0.17634651064872742
Step 501, loss: 0.17864353954792023
Step 502, loss: 0.17430740594863892
Step 503, loss: 0.17458714544773102
Step 504, loss: 0.17079102993011475
Step 505, loss: 0.1753505915403366
Step 506, loss: 0.17569056153297424
Step 507, loss: 0.17542502284049988
Step 508, loss: 0.17260128259658813
Step 509, loss: 0.1728348284959793
Step 510, loss: 0.16836656630039215
Step 511, loss: 0.1756671667098999
Step 512, loss: 0.17208218574523926
Step 513, loss: 0.17600437998771667
Step 514, loss: 0.16866090893745422
Step 515, loss: 0.17926707863807678
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 688, loss: 0.168025940656662
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 516, loss: 0.17697672545909882
Step 517, loss: 0.17678968608379364
Step 518, loss: 0.17956872284412384
Step 519, loss: 0.17370600998401642
Step 520, loss: 0.17517651617527008
Step 521, loss: 0.18297064304351807
Step 522, loss: 0.17328175902366638
Step 523, loss: 0.1723300963640213
Step 524, loss: 0.17295172810554504
Step 525, loss: 0.17543092370033264
Step 526, loss: 0.17585870623588562
Step 527, loss: 0.17599253356456757
Step 528, loss: 0.17656224966049194
Step 529, loss: 0.17330434918403625
Step 530, loss: 0.17364926636219025
Step 531, loss: 0.17312079668045044
Step 532, loss: 0.17132186889648438
Step 533, loss: 0.175296813249588
Step 534, loss: 0.17150579392910004
Step 535, loss: 0.17446093261241913
Step 536, loss: 0.1724652647972107
Step 537, loss: 0.1731881946325302
Step 538, loss: 0.1703382283449173
Step 539, loss: 0.17619144916534424
Step 540, loss: 0.17391523718833923
Step 541, loss: 0.1682422012090683
Step 542, loss: 0.1671500951051712
Step 543, loss: 0.17543551325798035
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 544, loss: 0.17303699254989624
Step 545, loss: 0.1722414791584015
Step 546, loss: 0.17534324526786804
Step 547, loss: 0.1776745766401291
Step 548, loss: 0.17317381501197815
Step 549, loss: 0.1715957671403885
Step 550, loss: 0.17733988165855408
Step 551, loss: 0.17435169219970703
Step 552, loss: 0.17444191873073578
Step 553, loss: 0.17420919239521027
Step 554, loss: 0.17075781524181366
Step 555, loss: 0.17201998829841614
Step 556, loss: 0.17111505568027496
Step 557, loss: 0.17426453530788422
Step 558, loss: 0.17481622099876404
Step 559, loss: 0.17275071144104004
Step 560, loss: 0.17486710846424103
Step 561, loss: 0.17817048728466034
Step 562, loss: 0.1738932728767395
Step 563, loss: 0.17490266263484955
Step 564, loss: 0.1721874177455902
Step 565, loss: 0.1751432716846466
Step 566, loss: 0.17336255311965942
Step 567, loss: 0.17610953748226166
Step 568, loss: 0.17619186639785767
Step 569, loss: 0.17549742758274078
Step 570, loss: 0.177082821726799
Step 571, loss: 0.17325030267238617
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 572, loss: 0.17269103229045868
Step 573, loss: 0.16923318803310394
Step 688, loss: 0.168025940656662
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 574, loss: 0.17252901196479797
Step 575, loss: 0.1739131659269333
Step 576, loss: 0.17262126505374908
Step 577, loss: 0.17376677691936493
Step 578, loss: 0.17164166271686554
Step 579, loss: 0.17398510873317719
Step 580, loss: 0.17701764404773712
Step 581, loss: 0.17249749600887299
Step 582, loss: 0.17545673251152039
Step 583, loss: 0.1745530217885971
Step 584, loss: 0.1704486757516861
Step 585, loss: 0.17150364816188812
Step 586, loss: 0.1709315925836563
Step 587, loss: 0.1745835691690445
Step 588, loss: 0.17887862026691437
Step 589, loss: 0.17313744127750397
Step 590, loss: 0.17206576466560364
Step 591, loss: 0.17699959874153137
Step 592, loss: 0.17157968878746033
Step 593, loss: 0.16830675303936005
Step 594, loss: 0.17418381571769714
Step 595, loss: 0.1751183122396469
Step 596, loss: 0.1756243109703064
Step 597, loss: 0.17006143927574158
Step 598, loss: 0.1699618548154831
Step 599, loss: 0.17193539440631866
Step 600, loss: 0.17631924152374268
Step 601, loss: 0.17033645510673523
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 602, loss: 0.1693136841058731
Step 603, loss: 0.17088831961154938
Step 604, loss: 0.17239458858966827
Step 605, loss: 0.17504796385765076
Step 606, loss: 0.17697462439537048
Step 607, loss: 0.17530332505702972
Step 608, loss: 0.17112182080745697
Step 609, loss: 0.1731833964586258
Step 610, loss: 0.17709961533546448
Step 611, loss: 0.17229171097278595
Step 612, loss: 0.17470520734786987
Step 613, loss: 0.1714342087507248
Step 614, loss: 0.17176499962806702
Step 615, loss: 0.17025373876094818
Step 616, loss: 0.174561008810997
Step 617, loss: 0.17275869846343994
Step 618, loss: 0.17185458540916443
Step 619, loss: 0.17359009385108948
Step 620, loss: 0.17360955476760864
Step 621, loss: 0.17254018783569336
Step 622, loss: 0.17021723091602325
Step 623, loss: 0.17180855572223663
Step 624, loss: 0.17233887314796448
Step 625, loss: 0.1753343939781189
Step 626, loss: 0.17674151062965393
Step 627, loss: 0.17399686574935913
Step 628, loss: 0.17107972502708435
Step 629, loss: 0.17392827570438385
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 630, loss: 0.1768183708190918
Step 631, loss: 0.1721939742565155
Step 632, loss: 0.16967353224754333
Step 633, loss: 0.17319253087043762
Step 634, loss: 0.17231595516204834
Step 635, loss: 0.17189306020736694
Step 636, loss: 0.1779624968767166
Step 637, loss: 0.1734601855278015
Step 638, loss: 0.17567768692970276
Step 639, loss: 0.17271453142166138
Step 640, loss: 0.1713656783103943
Step 641, loss: 0.17186495661735535
Step 642, loss: 0.1760798543691635
Step 643, loss: 0.17462579905986786
Step 644, loss: 0.17327368259429932
Step 645, loss: 0.1764570027589798
Step 646, loss: 0.17871925234794617
Step 647, loss: 0.17270927131175995
Step 648, loss: 0.175270676612854
Step 649, loss: 0.17492403090000153
Step 650, loss: 0.17410653829574585
Step 651, loss: 0.17691680788993835
Step 652, loss: 0.17220641672611237
Step 653, loss: 0.17202720046043396
Step 654, loss: 0.1755136102437973
Step 655, loss: 0.17424365878105164
Step 656, loss: 0.1737462878227234
Step 657, loss: 0.16914884746074677
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 658, loss: 0.17008645832538605
Step 659, loss: 0.1742316037416458
Step 660, loss: 0.17355559766292572
Step 661, loss: 0.16900521516799927
Step 662, loss: 0.17213037610054016
Step 663, loss: 0.1717776209115982
Step 664, loss: 0.1679377555847168
Step 665, loss: 0.1741892397403717
Step 666, loss: 0.1769883781671524
Step 667, loss: 0.16878068447113037
Step 668, loss: 0.1752023845911026
Step 669, loss: 0.17089705169200897
Step 670, loss: 0.1746937334537506
Step 671, loss: 0.17643389105796814
Step 672, loss: 0.1757017821073532
Step 673, loss: 0.1753089427947998
Step 674, loss: 0.175148144364357
Step 675, loss: 0.17286920547485352
Step 676, loss: 0.18160521984100342
Step 677, loss: 0.1749887466430664
Step 678, loss: 0.17353855073451996
Step 679, loss: 0.17348110675811768
Step 680, loss: 0.17679621279239655
Step 681, loss: 0.1688816398382187
Step 682, loss: 0.17490769922733307
Step 683, loss: 0.17182523012161255
Step 684, loss: 0.17375357449054718
Step 685, loss: 0.17375998198986053
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 686, loss: 0.1722177267074585
Step 687, loss: 0.17532184720039368
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 688, loss: 0.168025940656662
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 917, loss: 0.1942889243364334
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 917, loss: 0.1942889243364334
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 917, loss: 0.1942889243364334
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 917, loss: 0.1942889243364334
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 917, loss: 0.1942889243364334
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 917, loss: 0.1942889243364334
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 917, loss: 0.1942889243364334
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 917, loss: 0.1942889243364334
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 917, loss: 0.1942889243364334
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 917, loss: 0.1942889243364334
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 917, loss: 0.1942889243364334
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 917, loss: 0.1942889243364334
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 917, loss: 0.1942889243364334
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 917, loss: 0.1942889243364334
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 917, loss: 0.1942889243364334
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 917, loss: 0.1942889243364334
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 689, loss: 0.17899134755134583
Step 690, loss: 0.17709007859230042
Step 691, loss: 0.17748837172985077
Step 692, loss: 0.17005573213100433
Step 693, loss: 0.17669154703617096
Step 694, loss: 0.1746414601802826
Step 695, loss: 0.16808484494686127
Step 696, loss: 0.17436909675598145
Step 697, loss: 0.17107637226581573
Step 698, loss: 0.1729474663734436
Step 699, loss: 0.17476464807987213
Step 700, loss: 0.17165909707546234
Step 701, loss: 0.17324410378932953
Step 702, loss: 0.17189548909664154
Step 703, loss: 0.1744406521320343
Step 704, loss: 0.17370614409446716
Step 705, loss: 0.17620445787906647
Step 706, loss: 0.174727201461792
Step 707, loss: 0.17428751289844513
Step 708, loss: 0.17151211202144623
Step 709, loss: 0.1715441644191742
Step 710, loss: 0.17064987123012543
Step 711, loss: 0.17504172027111053
Step 712, loss: 0.17415207624435425
Step 713, loss: 0.17689689993858337
Step 714, loss: 0.1732635200023651
Step 715, loss: 0.17517362534999847
Step 716, loss: 0.1733883023262024
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 717, loss: 0.17322330176830292
Step 718, loss: 0.17422790825366974
Step 719, loss: 0.17621846497058868
Step 720, loss: 0.1731739491224289
Step 721, loss: 0.1730724275112152
Step 722, loss: 0.17115963995456696
Step 723, loss: 0.17356044054031372
Step 724, loss: 0.17796820402145386
Step 725, loss: 0.17152009904384613
Step 726, loss: 0.17454403638839722
Step 727, loss: 0.1752878725528717
Step 728, loss: 0.17705191671848297
Step 729, loss: 0.17480850219726562
Step 730, loss: 0.17321400344371796
Step 731, loss: 0.1733068823814392
Step 732, loss: 0.17345958948135376
Step 733, loss: 0.1765996515750885
Step 734, loss: 0.1709475815296173
Step 735, loss: 0.17511475086212158
Step 736, loss: 0.17417703568935394
Step 737, loss: 0.17453403770923615
Step 738, loss: 0.176243856549263
Step 739, loss: 0.17172643542289734
Step 740, loss: 0.17579515278339386
Step 741, loss: 0.17285293340682983
Step 742, loss: 0.17573365569114685
Step 743, loss: 0.17104139924049377
Step 744, loss: 0.1719278246164322
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 917, loss: 0.1942889243364334
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 745, loss: 0.16906842589378357
Step 746, loss: 0.17420704662799835
Step 747, loss: 0.17533841729164124
Step 748, loss: 0.17181938886642456
Step 749, loss: 0.17634889483451843
Step 750, loss: 0.17465147376060486
Step 751, loss: 0.1787661761045456
Step 752, loss: 0.17374488711357117
Step 753, loss: 0.17162586748600006
Step 754, loss: 0.1724761724472046
Step 755, loss: 0.17726847529411316
Step 756, loss: 0.17167915403842926
Step 757, loss: 0.17300309240818024
Step 758, loss: 0.172225683927536
Step 759, loss: 0.16727131605148315
Step 760, loss: 0.17071224749088287
Step 761, loss: 0.17563322186470032
Step 762, loss: 0.17502345144748688
Step 763, loss: 0.17246218025684357
Step 764, loss: 0.1696949303150177
Step 765, loss: 0.17512226104736328
Step 766, loss: 0.17293953895568848
Step 767, loss: 0.17304810881614685
Step 768, loss: 0.17117135226726532
Step 769, loss: 0.17588680982589722
Step 770, loss: 0.17896541953086853
Step 771, loss: 0.17421455681324005
Step 772, loss: 0.17155921459197998
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 773, loss: 0.1709890216588974
Step 774, loss: 0.1724115014076233
Step 775, loss: 0.1722429394721985
Step 776, loss: 0.1747157871723175
Step 777, loss: 0.1790749728679657
Step 778, loss: 0.1717798113822937
Step 779, loss: 0.1735348254442215
Step 780, loss: 0.17521053552627563
Step 781, loss: 0.17568501830101013
Step 782, loss: 0.17057661712169647
Step 783, loss: 0.17483651638031006
Step 784, loss: 0.17737214267253876
Step 785, loss: 0.17214278876781464
Step 786, loss: 0.1768963485956192
Step 787, loss: 0.1741919070482254
Step 788, loss: 0.17316724359989166
Step 789, loss: 0.17093606293201447
Step 790, loss: 0.175703763961792
Step 791, loss: 0.17170217633247375
Step 792, loss: 0.17211130261421204
Step 793, loss: 0.17337283492088318
Step 794, loss: 0.17829380929470062
Step 795, loss: 0.17962464690208435
Step 796, loss: 0.17689000070095062
Step 797, loss: 0.170718714594841
Step 798, loss: 0.1725749671459198
Step 799, loss: 0.17247985303401947
Step 800, loss: 0.17255917191505432
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 801, loss: 0.17473682761192322
Step 802, loss: 0.17221669852733612
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 803, loss: 0.16969819366931915
Step 804, loss: 0.17372442781925201
Step 805, loss: 0.16981947422027588
Step 806, loss: 0.17430536448955536
Step 807, loss: 0.1727956086397171
Step 808, loss: 0.1761023849248886
Step 809, loss: 0.17369487881660461
Step 810, loss: 0.17208799719810486
Step 811, loss: 0.17292466759681702
Step 812, loss: 0.17395928502082825
Step 813, loss: 0.17069312930107117
Step 814, loss: 0.1750108152627945
Step 815, loss: 0.17069949209690094
Step 816, loss: 0.18419882655143738
Step 817, loss: 0.1798088252544403
Step 818, loss: 0.18515458703041077
Step 819, loss: 0.18215692043304443
Step 820, loss: 0.19395676255226135
Step 821, loss: 0.18590819835662842
Step 822, loss: 0.18910640478134155
Step 823, loss: 0.19816628098487854
Step 824, loss: 0.1925753653049469
Step 825, loss: 0.19668900966644287
Step 826, loss: 0.1993279755115509
Step 827, loss: 0.19620366394519806
Step 828, loss: 0.19355981051921844
Step 829, loss: 0.19763712584972382
Step 830, loss: 0.19579480588436127
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 831, loss: 0.19501915574073792
Step 832, loss: 0.19334354996681213
Step 833, loss: 0.194542795419693
Step 834, loss: 0.19733156263828278
Step 835, loss: 0.19719944894313812
Step 836, loss: 0.19262580573558807
Step 837, loss: 0.19488126039505005
Step 838, loss: 0.20572589337825775
Step 839, loss: 0.19548571109771729
Step 840, loss: 0.19498057663440704
Step 841, loss: 0.19680288434028625
Step 842, loss: 0.19567465782165527
Step 843, loss: 0.19134315848350525
Step 844, loss: 0.19077661633491516
Step 845, loss: 0.19448570907115936
Step 846, loss: 0.19932132959365845
Step 847, loss: 0.1971946656703949
Step 848, loss: 0.19468267261981964
Step 849, loss: 0.1999991238117218
Step 850, loss: 0.19588442146778107
Step 851, loss: 0.19512861967086792
Step 852, loss: 0.19713445007801056
Step 853, loss: 0.19512681663036346
Step 854, loss: 0.19677601754665375
Step 855, loss: 0.19457361102104187
Step 856, loss: 0.1932019740343094
Step 857, loss: 0.19812166690826416
Step 858, loss: 0.1941109299659729
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 859, loss: 0.1927156299352646
Step 860, loss: 0.1943768411874771
Step 861, loss: 0.19775605201721191
Step 862, loss: 0.19254070520401
Step 863, loss: 0.19562996923923492
Step 864, loss: 0.1940685212612152
Step 865, loss: 0.19106583297252655
Step 866, loss: 0.19553005695343018
Step 867, loss: 0.19323907792568207
Step 868, loss: 0.1959063708782196
Step 869, loss: 0.1980343759059906
Step 870, loss: 0.19228437542915344
Step 871, loss: 0.19487135112285614
Step 872, loss: 0.19422484934329987
Step 873, loss: 0.19584865868091583
Step 874, loss: 0.19885213673114777
Step 875, loss: 0.19261327385902405
Step 876, loss: 0.19618886709213257
Step 877, loss: 0.19166012108325958
Step 878, loss: 0.19807542860507965
Step 879, loss: 0.19533373415470123
Step 880, loss: 0.19554272294044495
Step 881, loss: 0.1983797401189804
Step 882, loss: 0.19314245879650116
Step 883, loss: 0.1953534185886383
Step 884, loss: 0.19479385018348694
Step 885, loss: 0.19552813470363617
Step 886, loss: 0.19319650530815125
Step 917, loss: 0.1942889243364334
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 917, loss: 0.1942889243364334
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 887, loss: 0.19092334806919098
Step 888, loss: 0.19556617736816406
Step 889, loss: 0.19262593984603882
Step 890, loss: 0.19435875117778778
Step 891, loss: 0.1943531036376953
Step 892, loss: 0.19408383965492249
Step 893, loss: 0.1942259818315506
Step 894, loss: 0.19630619883537292
Step 895, loss: 0.19281530380249023
Step 896, loss: 0.19543924927711487
Step 897, loss: 0.2001473754644394
Step 898, loss: 0.19750091433525085
Step 899, loss: 0.195859894156456
Step 900, loss: 0.19264118373394012
Step 901, loss: 0.19377319514751434
Step 902, loss: 0.19348706305027008
Step 903, loss: 0.19239738583564758
Step 904, loss: 0.19326980412006378
Step 905, loss: 0.1943923681974411
Step 906, loss: 0.1941441297531128
Step 907, loss: 0.19398580491542816
Step 908, loss: 0.1948879212141037
Step 909, loss: 0.1920291930437088
Step 910, loss: 0.19403284788131714
Step 911, loss: 0.1945066601037979
Step 912, loss: 0.197319895029068
Step 913, loss: 0.19126680493354797
Step 914, loss: 0.19808724522590637
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 917, loss: 0.1942889243364334
Step 915, loss: 0.19713211059570312
Step 916, loss: 0.19838380813598633
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 917, loss: 0.1942889243364334
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=11] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=9] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=8] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
ValueError: [process_index=10] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
ValueError: [process_index=25] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=5] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=27] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=7] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=4] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
ValueError: [process_index=24] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=6] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=26] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318099_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Step 918, loss: 0.19523441791534424
Step 919, loss: 0.19546496868133545
Step 920, loss: 0.19409704208374023
Step 921, loss: 0.19166922569274902
Step 922, loss: 0.1975981593132019
Step 923, loss: 0.19526618719100952
Step 924, loss: 0.19312119483947754
Step 925, loss: 0.19502730667591095
Step 926, loss: 0.19699089229106903
Step 927, loss: 0.19339591264724731
Step 928, loss: 0.19219785928726196
Step 929, loss: 0.19684770703315735
Step 930, loss: 0.1927916556596756
Step 931, loss: 0.1985294222831726
Step 932, loss: 0.1954779475927353
Step 933, loss: 0.19568152725696564
Step 934, loss: 0.19383743405342102
Step 935, loss: 0.19561338424682617
Step 936, loss: 0.19460172951221466
Step 937, loss: 0.19666717946529388
Step 938, loss: 0.1971161663532257
Step 939, loss: 0.19739972054958344
Step 940, loss: 0.19577215611934662
Step 941, loss: 0.19569700956344604
Step 942, loss: 0.1939944624900818
Step 943, loss: 0.1971275806427002
Step 944, loss: 0.19501261413097382
Step 945, loss: 0.19413445889949799
Step 946, loss: 0.1940343976020813
Step 947, loss: 0.19624318182468414
Step 948, loss: 0.19629356265068054
Step 949, loss: 0.19340308010578156
Step 950, loss: 0.193202942609787
Step 951, loss: 0.19428031146526337
Step 952, loss: 0.19374477863311768
Step 953, loss: 0.1919095665216446
Step 954, loss: 0.19433891773223877
Step 955, loss: 0.19187457859516144
Step 956, loss: 0.19621175527572632
Step 957, loss: 0.19424453377723694
Step 958, loss: 0.19666950404644012
Step 959, loss: 0.19401375949382782
Step 960, loss: 0.19302953779697418
Step 961, loss: 0.19898679852485657
Step 962, loss: 0.19154222309589386
Step 963, loss: 0.1965298056602478
Step 964, loss: 0.19366717338562012
Step 965, loss: 0.19513016939163208
Step 966, loss: 0.19310520589351654
Step 967, loss: 0.19546397030353546
Step 968, loss: 0.19591833651065826
Step 969, loss: 0.19176341593265533
Step 970, loss: 0.19398139417171478
Step 971, loss: 0.19647134840488434
Step 972, loss: 0.1946771889925003
Step 973, loss: 0.19320639967918396
Step 974, loss: 0.19349446892738342
Step 975, loss: 0.19277448952198029
Step 976, loss: 0.19457297027111053
Step 977, loss: 0.19542548060417175
Step 978, loss: 0.19564610719680786
Step 979, loss: 0.19077078998088837
Step 980, loss: 0.19871804118156433
Step 981, loss: 0.1972716599702835
Step 982, loss: 0.19280266761779785
Step 983, loss: 0.19526249170303345
Step 984, loss: 0.19640924036502838
Step 985, loss: 0.19558152556419373
Step 986, loss: 0.19950821995735168
Step 987, loss: 0.19390521943569183
Step 988, loss: 0.19487619400024414
Step 989, loss: 0.19438578188419342
Step 990, loss: 0.19506306946277618
Step 991, loss: 0.19303816556930542
Step 992, loss: 0.19323404133319855
Step 993, loss: 0.1923423707485199
Step 994, loss: 0.1945382058620453
Step 995, loss: 0.19345350563526154
Step 996, loss: 0.1991584599018097
Step 997, loss: 0.19994023442268372
Step 998, loss: 0.1919410675764084
Step 999, loss: 0.19577297568321228
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
ValueError: [process_index=23] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318122_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=16] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318118_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=18] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318118_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
ValueError: [process_index=31] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318120_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=21] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318122_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=14] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318123_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
ValueError: [process_index=12] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318123_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=17] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318118_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
ValueError: [process_index=28] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318120_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=20] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318122_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
    await asyncio.sleep(0.25)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return fut.result()
asyncio.exceptions.CancelledError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    await asyncio.wait_for(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/train_tokenizer.py", line 262, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
ValueError: [process_index=19] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318118_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    return self._f.result(timeout=timeout)
ValueError: [process_index=22] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318122_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=13] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318123_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=15] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318123_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=30] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318120_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._result = target()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-pai00039/tum_ind3695/.local/share/uv/python/cpython-3.10.17-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-pai00039/tum_ind3695/projects/jafar_batch_size_scaling/.venv_jafar/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    raise ValueError(
ValueError: [process_index=29] Timed out waiting for array_metadatas base directory creation: /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/tokenizer/train_tokenizer_batch_size_scaling_8_node_3307474/tokenizer_1751318120_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
2025-07-01 00:12:31.357123: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service.cc:1689] Shutdown barrier in coordination service has failed:
DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']
2025-07-01 00:12:31.357635: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
This suggests that the workers are out of sync. Either at least one worker (a) crashed early due to program error or scheduler events (e.g. preemption, eviction), (b) was too fast in its execution, or (c) too slow / hanging. Check the logs (both the program and scheduler events) for an earlier error to identify the root cause.
2025-07-01 00:12:31.357150: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service.cc:1730] Use error polling to propagate the following error to all tasks: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
2025-07-01 00:12:31.357272: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357408: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
2025-07-01 00:12:31.357511: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380'] [type.googleapis.com/tensorflow.CoordinationServiceError='']
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.357537: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357838: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
2025-07-01 00:12:31.357879: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357487: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357789: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.358027: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.357624: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.357937: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.357864: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.357965: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.358050: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357759: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357856: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.358181: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.357705: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.358251: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.357880: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.357838: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.357950: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
2025-07-01 00:12:31.358247: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
2025-07-01 00:12:31.358561: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.358682: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.357944: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.358160: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

2025-07-01 00:12:31.358277: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.358191: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388132: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.389752: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.387984: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.389659: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388310: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.389838: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388302: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.389938: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388225: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390670: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
2025-07-01 00:12:31.374109: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390981: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388191: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390626: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
2025-07-01 00:12:31.374262: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.391021: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388298: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390927: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
2025-07-01 00:12:31.374546: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.391311: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.387946: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390776: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388298: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390889: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.374395: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.391196: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388167: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390957: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388074: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.390919: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-01 00:12:31.388180: E external/xla/xla/tsl/distributed_runtime/coordination/coordination_service_agent.cc:427] Polled an error from coordination service (this can be an error from this or another task).
2025-07-01 00:12:31.391065: F external/xla/xla/pjrt/distributed/client.h:80] Terminating process because the JAX distributed service detected fatal errors. This most likely indicates that another task died; see the other task logs for more details. Disable Python buffering, i.e. `python -u`, to be sure to see all the previous output. absl::Status: INTERNAL: Failed to disconnect from coordination service with status: INTERNAL: Shutdown barrier has failed.
Barrier result: 'DEADLINE_EXCEEDED: Barrier timed out. Id: Shutdown::16634615291091153380::0. This usually happens because a task triggered the barrier too early or too slowly. Please look at the task logs (both timed out and first task) to debug further.
# of tasks that reached the barrier: 28/32.
The first task at the barrier: /job:jax_worker/replica:0/task:5. Some timed out task names:
/job:jax_worker/replica:0/task:2
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16634615291091153380']

RPC: /tensorflow.CoordinationService/ShutdownTask [type.googleapis.com/tensorflow.CoordinationServiceError='']Proceeding with agent shutdown anyway. This is usually caused by an earlier error during execution. Check the logs of (a) this task, (b) the leader (usually slice 0 task 0) and (c) the scheduler (e.g. preemption, eviction) for an earlier error to debug further. [type.googleapis.com/tensorflow.CoordinationServiceError='']
srun: error: hkn0520: task 10: Aborted (core dumped)
srun: error: hkn0523: task 15: Aborted (core dumped)
srun: error: hkn0820: task 31: Aborted (core dumped)
srun: error: hkn0810: task 27: Aborted (core dumped)
srun: error: hkn0420: tasks 1-3: Aborted (core dumped)
srun: error: hkn0516: tasks 5,7: Aborted (core dumped)
srun: error: hkn0717: tasks 20,22: Aborted (core dumped)
srun: error: hkn0520: task 11: Aborted (core dumped)
srun: error: hkn0704: tasks 17,19: Aborted (core dumped)
srun: error: hkn0820: task 29: Aborted (core dumped)
srun: error: hkn0810: task 24: Aborted (core dumped)
srun: error: hkn0520: task 9: Aborted (core dumped)
srun: error: hkn0820: task 28: Aborted (core dumped)
srun: error: hkn0516: task 4: Aborted (core dumped)
srun: error: hkn0704: task 16: Aborted (core dumped)
srun: error: hkn0523: tasks 13-14: Aborted (core dumped)
srun: error: hkn0717: task 21: Aborted (core dumped)
srun: error: hkn0717: task 23: Aborted (core dumped)
srun: error: hkn0820: task 30: Aborted (core dumped)
srun: error: hkn0810: task 26: Aborted (core dumped)
srun: error: hkn0520: task 8: Aborted (core dumped)
srun: error: hkn0516: task 6: Aborted (core dumped)
srun: error: hkn0810: task 25: Aborted (core dumped)
srun: error: hkn0523: task 12: Aborted (core dumped)
srun: error: hkn0704: task 18: Aborted (core dumped)
srun: error: hkn0420: task 0: Aborted (core dumped)

============================= JOB FEEDBACK =============================

Job ID: 3307474
Cluster: hk
User/Group: tum_ind3695/hk-project-pai00039
Account: hk-project-p0023960
State: FAILED (exit code 134)
Partition: accelerated
Nodes: 8
Cores per node: 32
Nodelist: hkn[0420,0516,0520,0523,0704,0717,0810,0820]
CPU Utilized: 22:56:17
CPU Efficiency: 9.24% of 10-08:10:40 core-walltime
Job Wall-clock time: 00:58:10
Starttime: Mon Jun 30 23:14:23 2025
Endtime: Tue Jul  1 00:12:33 2025
Memory Utilized: 423.14 GB (estimated maximum)
Memory Efficiency: 0.00% of 0.00 MB (0.00 MB/node)
Energy Consumed: 35390279 Joule / 9830.63305555556 Watthours
Average node power draw: 10140.4810888252 Watt