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

#SBATCH --nodes=4
#SBATCH --ntasks-per-node=4
#SBATCH --time=36:00:00
#SBATCH --partition=accelerated
#SBATCH --cpus-per-task=5
#SBATCH --gres=gpu:4
#SBATCH --output=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/%x_%j.log
#SBATCH --error=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/%x_%j.log
#SBATCH --job-name=train_tokenizer_model_size_scaling_74M
#SBATCH --constraint=BEEOND

# Log the sbatch script
cat $0

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

tf_records_dir=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/data/open_ai_minecraft_tfrecord

job_name=$SLURM_JOB_NAME
slurm_job_id=$SLURM_JOB_ID

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

ODFSDIR=/mnt/odfs/$SLURM_JOB_ID/stripe_default/checkpoints/
mkdir -p $ODFSDIR

env | grep SLURM

srun python train_tokenizer.py \
    --ckpt_dir $CHECKPOINT_DIR \
    --tmp_ckpt_dir $ODFSDIR \
    --batch_size=96 \
    --min_lr=0 \
    --max_lr=7e-5 \
    --log_image_interval=500 \
    --log_checkpoint_interval=1000 \
    --log \
    --name=tokenizer-modelsize-74M-$slurm_job_id \
    --tags tokenizer model-size-scaling 74M \
    --entity instant-uv \
    --project jafar \
    --data_dir $tf_records_dir \
    --model_dim=640 \
    --num_blocks=10 \
    --num_heads=10 \
    --latent_dim=48 \
    --num_latents=1536
/var/spool/slurmd/job3318557/slurm_script: line 19: .venv/bin/activate: No such file or directory
SLURM_JOB_USER=tum_cte0515
SLURM_TASKS_PER_NODE=4(x4)
SLURM_JOB_UID=999226
SLURM_TASK_PID=2034614
SLURM_JOB_GPUS=0,1,2,3
SLURM_LOCALID=0
SLURM_SUBMIT_DIR=/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar_jobs
SLURMD_NODENAME=hkn0414
SLURM_JOB_START_TIME=1751679369
SLURM_CLUSTER_NAME=hk
SLURM_JOB_END_TIME=1751808969
SLURM_CPUS_ON_NODE=24
SLURM_JOB_CPUS_PER_NODE=24(x4)
SLURM_GPUS_ON_NODE=4
SLURM_GTIDS=0
SLURM_JOB_PARTITION=accelerated
SLURM_TRES_PER_TASK=cpu=5
SLURM_OOM_KILL_STEP=0
SLURM_JOB_NUM_NODES=4
SLURM_JOBID=3318557
SLURM_JOB_QOS=normal
SLURM_PROCID=0
SLURM_CPUS_PER_TASK=5
SLURM_NTASKS=16
SLURM_TOPOLOGY_ADDR=hkibb.hkibbi1.hkibbi1e12.hkn0414
SLURM_TOPOLOGY_ADDR_PATTERN=switch.switch.switch.node
SLURM_SCRIPT_CONTEXT=prolog_task
SLURM_NODELIST=hkn[0414,0416,0511,0519]
SLURM_JOB_ACCOUNT=hk-project-p0023960
SLURM_PRIO_PROCESS=0
SLURM_NPROCS=16
SLURM_NNODES=4
SLURM_SUBMIT_HOST=hkn1993.localdomain
SLURM_JOB_ID=3318557
SLURM_NODEID=0
SLURM_CONF=/etc/slurm/slurm.conf
SLURM_JOB_NAME=train_tokenizer_model_size_scaling_74M
SLURM_NTASKS_PER_NODE=4
SLURM_JOB_GID=502226
SLURM_JOB_NODELIST=hkn[0414,0416,0511,0519]
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
GpuFreq=control_disabled
2025-07-05 03:37:27.289652: 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-07-05 03:37:27.289812: 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-07-05 03:37:27.289919: 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-07-05 03:37:27.290334: 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:1751679447.303101  187634 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:1751679447.303352  187632 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:1751679447.303227  187633 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:1751679447.303410  187631 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:1751679447.307391  187633 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:1751679447.307766  187634 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:1751679447.308116  187632 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:1751679447.308258  187631 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:1751679447.320292  187633 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.320309  187633 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.320311  187633 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.320312  187633 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321757  187634 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321772  187634 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321775  187634 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321776  187634 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321854  187631 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321868  187631 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321869  187631 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.321871  187631 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.322146  187632 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.322160  187632 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.322161  187632 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679447.322163  187632 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-07-05 03:37:31.448935: 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-07-05 03:37:31.449211: 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-07-05 03:37:31.449849: 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-07-05 03:37:31.449876: 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:1751679451.462273 3586910 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:1751679451.462821 3586909 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:1751679451.462919 3586912 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:1751679451.462960 3586911 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:1751679451.466661 3586910 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:1751679451.467128 3586911 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:1751679451.467080 3586912 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:1751679451.467183 3586909 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:1751679451.480678 3586910 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480694 3586910 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480696 3586910 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480698 3586910 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480685 3586911 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480704 3586911 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480705 3586911 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480707 3586911 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480689 3586912 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480703 3586912 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480705 3586912 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480707 3586912 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480828 3586909 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480847 3586909 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480849 3586909 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.480850 3586909 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-07-05 03:37:31.733759: 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-07-05 03:37:31.733747: 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-07-05 03:37:31.733978: 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-07-05 03:37:31.734249: 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:1751679451.747058  106409 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:1751679451.747076  106410 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:1751679451.747061  106411 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:1751679451.747269  106412 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:1751679451.751184  106411 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:1751679451.751357  106409 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:1751679451.751539  106410 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:1751679451.751596  106412 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:1751679451.764418  106411 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.764435  106411 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.764437  106411 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.764439  106411 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765013  106412 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765030  106412 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765031  106412 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765033  106412 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765159  106410 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765174  106410 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765176  106410 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765177  106410 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765414  106409 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765431  106409 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765432  106409 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679451.765434  106409 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
2025-07-05 03:37:32.993479: 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-07-05 03:37:32.993476: 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-07-05 03:37:32.993472: 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-07-05 03:37:32.993483: 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:1751679453.090247 2034741 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:1751679453.090243 2034742 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:1751679453.090163 2034743 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:1751679453.090329 2034744 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:1751679453.125642 2034742 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:1751679453.125591 2034744 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:1751679453.125722 2034743 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:1751679453.125814 2034741 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:1751679453.479145 2034741 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479171 2034741 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479174 2034741 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479176 2034741 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479145 2034742 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479174 2034742 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479177 2034742 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479178 2034742 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479149 2034743 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479179 2034743 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479181 2034743 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479182 2034743 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479153 2034744 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479176 2034744 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479178 2034744 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679453.479179 2034744 computation_placer.cc:177] computation placer already registered. Please check linkage and avoid linking the same target more than once.
W0000 00:00:1751679466.383200  187631 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:1751679466.383199  187632 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:1751679466.383188  187633 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:1751679466.383215  187634 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:1751679472.819154  106410 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:1751679472.820373  106409 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:1751679472.820354  106412 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:1751679472.820475  106411 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:1751679472.885510 3586909 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:1751679472.885494 3586911 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:1751679472.885757 3586910 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:1751679472.885686 3586912 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:1751679473.473389 2034741 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:1751679473.473387 2034743 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:1751679473.473453 2034744 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:1751679473.473539 2034742 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-07-05 03:37:58.919584: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.919487: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.921040: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.932138: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.934091: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.937126: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.939881: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.946306: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.951010: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.951959: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.952806: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:58.963285: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:59.190050: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:59.191953: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:59.201796: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:37:59.201900: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
wandb: Currently logged in as: mihir-mahajan2002 (instant-uv) to https://api.wandb.ai. Use `wandb login --relogin` to force relogin
wandb: Tracking run with wandb version 0.19.11
wandb: Run data is saved locally in /hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar_jobs/wandb/run-20250705_033818-4ght8phr
wandb: Run `wandb offline` to turn off syncing.
wandb: Syncing run tokenizer-modelsize-74M-3318557
wandb: ⭐️ View project at https://wandb.ai/instant-uv/jafar
wandb: 🚀 View run at https://wandb.ai/instant-uv/jafar/runs/4ght8phr
2025-07-05 03:38:52.502065: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:38:55.087023: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
2025-07-05 03:38:55.237702: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 111, loss: 0.15413720905780792
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 111, loss: 0.15413720905780792
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 111, loss: 0.15413720905780792
Step 111, loss: 0.15413720905780792
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 111, loss: 0.15413720905780792
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 111, loss: 0.15413720905780792
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 111, loss: 0.15413720905780792
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 111, loss: 0.15413720905780792
Step 111, loss: 0.15413720905780792
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 111, loss: 0.15413720905780792
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 111, loss: 0.15413720905780792
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 111, loss: 0.15413720905780792
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 111, loss: 0.15413720905780792
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Running on 16 devices.
Counting all components: ['encoder', 'vq', 'decoder']
Parameter counts:
{'encoder': 37023504, 'vq': 73728, 'decoder': 37023504, 'total': 74120736}
Starting training from step 0...
Step 0, loss: 0.30343344807624817
Step 1, loss: 0.29895898699760437
Step 2, loss: 0.29286086559295654
Step 3, loss: 0.30027419328689575
Step 4, loss: 0.2998599708080292
Step 5, loss: 0.3030512034893036
Step 6, loss: 0.3046490550041199
Step 7, loss: 0.30385923385620117
Step 8, loss: 0.29523614048957825
Step 9, loss: 0.2936331331729889
Step 10, loss: 0.29248130321502686
Step 11, loss: 0.2879736125469208
Step 12, loss: 0.2997485101222992
Step 13, loss: 0.2909526824951172
Step 14, loss: 0.29787614941596985
Step 15, loss: 0.29668310284614563
Step 16, loss: 0.2979995310306549
Step 17, loss: 0.302659273147583
Step 18, loss: 0.3138748109340668
Step 19, loss: 0.2962881922721863
Step 20, loss: 0.2866121232509613
Step 21, loss: 0.29632753133773804
Step 22, loss: 0.2969650626182556
Step 23, loss: 0.2962903082370758
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 24, loss: 0.2928117513656616
Step 25, loss: 0.28961533308029175
Step 26, loss: 0.29517078399658203
Step 27, loss: 0.28963908553123474
Step 28, loss: 0.29196617007255554
Step 29, loss: 0.27965304255485535
Step 30, loss: 0.2841093838214874
Step 31, loss: 0.2789527475833893
Step 32, loss: 0.2848070561885834
Step 33, loss: 0.28251147270202637
Step 34, loss: 0.2708398401737213
Step 35, loss: 0.29008346796035767
Step 36, loss: 0.28284192085266113
Step 37, loss: 0.2652813792228699
Step 38, loss: 0.2804304361343384
Step 39, loss: 0.2765401005744934
Step 40, loss: 0.28431016206741333
Step 41, loss: 0.27292561531066895
Step 42, loss: 0.28424760699272156
Step 43, loss: 0.27214765548706055
Step 44, loss: 0.2752127945423126
Step 45, loss: 0.2610222399234772
Step 46, loss: 0.2597903311252594
Step 47, loss: 0.26959022879600525
Step 48, loss: 0.26234397292137146
Step 49, loss: 0.25911253690719604
Step 50, loss: 0.2608296871185303
Step 51, loss: 0.25693053007125854
Step 52, loss: 0.2468448281288147
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 111, loss: 0.15413720905780792
Step 53, loss: 0.25405552983283997
Step 54, loss: 0.2572475075721741
Step 55, loss: 0.2519712746143341
Step 56, loss: 0.24430017173290253
Step 57, loss: 0.23902329802513123
Step 58, loss: 0.24360182881355286
Step 59, loss: 0.23382985591888428
Step 60, loss: 0.2311876118183136
Step 61, loss: 0.23892879486083984
Step 62, loss: 0.2431456744670868
Step 63, loss: 0.22652563452720642
Step 64, loss: 0.2273309826850891
Step 65, loss: 0.23480546474456787
Step 66, loss: 0.23718653619289398
Step 67, loss: 0.23304331302642822
Step 68, loss: 0.22457820177078247
Step 69, loss: 0.22480599582195282
Step 70, loss: 0.221456840634346
Step 71, loss: 0.2192995846271515
Step 72, loss: 0.21774634718894958
Step 73, loss: 0.21963317692279816
Step 74, loss: 0.21851909160614014
Step 75, loss: 0.21726851165294647
Step 76, loss: 0.216159850358963
Step 77, loss: 0.20812872052192688
Step 78, loss: 0.2114444077014923
Step 79, loss: 0.20741882920265198
Step 80, loss: 0.20693671703338623
Step 81, loss: 0.20144341886043549
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 82, loss: 0.2026442587375641
Step 83, loss: 0.20347024500370026
Step 84, loss: 0.1979225128889084
Step 85, loss: 0.20389962196350098
Step 86, loss: 0.19218449294567108
Step 87, loss: 0.19724713265895844
Step 88, loss: 0.19027340412139893
Step 89, loss: 0.19230493903160095
Step 90, loss: 0.18717285990715027
Step 91, loss: 0.18733014166355133
Step 92, loss: 0.18383461236953735
Step 93, loss: 0.19191673398017883
Step 94, loss: 0.18306954205036163
Step 95, loss: 0.1787676215171814
Step 96, loss: 0.1767098754644394
Step 97, loss: 0.17496849596500397
Step 98, loss: 0.17481227219104767
Step 99, loss: 0.18404221534729004
Step 100, loss: 0.17264342308044434
Step 101, loss: 0.1762748509645462
Step 102, loss: 0.16048112511634827
Step 103, loss: 0.18348102271556854
Step 104, loss: 0.16717667877674103
Step 105, loss: 0.17232611775398254
Step 106, loss: 0.16678571701049805
Step 107, loss: 0.16720527410507202
Step 108, loss: 0.16417060792446136
Step 109, loss: 0.16646942496299744
Step 110, loss: 0.16661298274993896
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 111, loss: 0.15413720905780792
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 111, loss: 0.15413720905780792
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 112, loss: 0.15752020478248596
Step 113, loss: 0.15663903951644897
Step 114, loss: 0.1647176295518875
Step 115, loss: 0.15767288208007812
Step 116, loss: 0.15905725955963135
Step 117, loss: 0.16607415676116943
Step 118, loss: 0.15268662571907043
Step 119, loss: 0.1383640617132187
Step 120, loss: 0.14784401655197144
Step 121, loss: 0.1350029557943344
Step 122, loss: 0.14504872262477875
Step 123, loss: 0.1400650292634964
Step 124, loss: 0.13831141591072083
Step 125, loss: 0.14437779784202576
Step 126, loss: 0.14216719567775726
Step 127, loss: 0.14621375501155853
Step 128, loss: 0.1379225105047226
Step 129, loss: 0.12637203931808472
Step 130, loss: 0.13955402374267578
Step 131, loss: 0.12941916286945343
Step 132, loss: 0.13250745832920074
Step 133, loss: 0.12927155196666718
Step 134, loss: 0.1347314566373825
Step 135, loss: 0.12530621886253357
Step 136, loss: 0.12599524855613708
Step 137, loss: 0.12620054185390472
Step 138, loss: 0.1361616551876068
Step 139, loss: 0.13306696712970734
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 140, loss: 0.12823618948459625
Step 141, loss: 0.1295480877161026
Step 142, loss: 0.1291031539440155
Step 143, loss: 0.12036333233118057
Step 144, loss: 0.1279449164867401
Step 145, loss: 0.1269577294588089
Step 146, loss: 0.12473965436220169
Step 147, loss: 0.12872476875782013
Step 148, loss: 0.12918174266815186
Step 149, loss: 0.1308222860097885
Step 150, loss: 0.12247758358716965
Step 151, loss: 0.11620639264583588
Step 152, loss: 0.1171131581068039
Step 153, loss: 0.12727157771587372
Step 154, loss: 0.12056201696395874
Step 155, loss: 0.1127057671546936
Step 156, loss: 0.12547510862350464
Step 157, loss: 0.12416236847639084
Step 158, loss: 0.12331140786409378
Step 159, loss: 0.11958343535661697
Step 160, loss: 0.11601174622774124
Step 161, loss: 0.10385622829198837
Step 162, loss: 0.10344018042087555
Step 163, loss: 0.12390053272247314
Step 164, loss: 0.1069401204586029
Step 165, loss: 0.11296728253364563
Step 166, loss: 0.11608634144067764
Step 167, loss: 0.10905099660158157
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 168, loss: 0.1162833496928215
Step 169, loss: 0.10483333468437195
Step 170, loss: 0.1039208471775055
Step 171, loss: 0.1069473922252655
Step 172, loss: 0.11145742982625961
Step 173, loss: 0.11640840023756027
Step 174, loss: 0.11411542445421219
Step 175, loss: 0.10890141874551773
Step 176, loss: 0.11100514978170395
Step 177, loss: 0.10419625788927078
Step 178, loss: 0.10314763337373734
Step 179, loss: 0.10528206080198288
Step 180, loss: 0.11151131242513657
Step 181, loss: 0.09695274382829666
Step 182, loss: 0.10219097137451172
Step 183, loss: 0.0993114709854126
Step 184, loss: 0.11306460946798325
Step 185, loss: 0.10966514050960541
Step 186, loss: 0.11388102918863297
Step 187, loss: 0.10111595690250397
Step 188, loss: 0.11295414716005325
Step 189, loss: 0.11512050032615662
Step 190, loss: 0.09318426996469498
Step 191, loss: 0.1033499613404274
Step 192, loss: 0.10239195078611374
Step 193, loss: 0.10757463425397873
Step 194, loss: 0.11065462231636047
Step 195, loss: 0.09807135909795761
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 196, loss: 0.09908454120159149
Step 197, loss: 0.09251553565263748
Step 198, loss: 0.09192368388175964
Step 199, loss: 0.09808102250099182
Step 200, loss: 0.09400425851345062
Step 201, loss: 0.0991429015994072
Step 202, loss: 0.0884544774889946
Step 203, loss: 0.0906473845243454
Step 204, loss: 0.08684232831001282
Step 205, loss: 0.08219499886035919
Step 206, loss: 0.09435367584228516
Step 207, loss: 0.08558663725852966
Step 208, loss: 0.09238786995410919
Step 209, loss: 0.08473894745111465
Step 210, loss: 0.08373972773551941
Step 211, loss: 0.08324525505304337
Step 212, loss: 0.09002130478620529
Step 213, loss: 0.08161056041717529
Step 214, loss: 0.08201031386852264
Step 215, loss: 0.07282457500696182
Step 216, loss: 0.0837293192744255
Step 217, loss: 0.0799691379070282
Step 218, loss: 0.08899373561143875
Step 219, loss: 0.09409746527671814
Step 220, loss: 0.07467888295650482
Step 221, loss: 0.0797310322523117
Step 222, loss: 0.09172932058572769
Step 223, loss: 0.08267998695373535
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 224, loss: 0.0846037045121193
Step 225, loss: 0.08669281750917435
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 454, loss: 0.06368210911750793
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 454, loss: 0.06368210911750793
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 454, loss: 0.06368210911750793
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 454, loss: 0.06368210911750793
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 454, loss: 0.06368210911750793
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 454, loss: 0.06368210911750793
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 454, loss: 0.06368210911750793
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 454, loss: 0.06368210911750793
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 454, loss: 0.06368210911750793
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 454, loss: 0.06368210911750793
Step 454, loss: 0.06368210911750793
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 226, loss: 0.0786675363779068
Step 227, loss: 0.08253331482410431
Step 228, loss: 0.07562312483787537
Step 229, loss: 0.07742524892091751
Step 230, loss: 0.0785994604229927
Step 231, loss: 0.07074376940727234
Step 232, loss: 0.08390665054321289
Step 233, loss: 0.07376068830490112
Step 234, loss: 0.07340297847986221
Step 235, loss: 0.0728844404220581
Step 236, loss: 0.08376455307006836
Step 237, loss: 0.07621903717517853
Step 238, loss: 0.07701282203197479
Step 239, loss: 0.07603820413351059
Step 240, loss: 0.06758818030357361
Step 241, loss: 0.07443436235189438
Step 242, loss: 0.0756043866276741
Step 243, loss: 0.07815448939800262
Step 244, loss: 0.07562774419784546
Step 245, loss: 0.0768357589840889
Step 246, loss: 0.072440505027771
Step 247, loss: 0.07444529980421066
Step 248, loss: 0.0747969001531601
Step 249, loss: 0.07463864237070084
Step 250, loss: 0.08198733627796173
Step 251, loss: 0.0787014439702034
Step 252, loss: 0.07338882982730865
Step 253, loss: 0.0794706717133522
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 254, loss: 0.07680429518222809
Step 255, loss: 0.07421426475048065
Step 256, loss: 0.08802162110805511
Step 257, loss: 0.0899503231048584
Step 258, loss: 0.08080412447452545
Step 259, loss: 0.07679661363363266
Step 260, loss: 0.07676488906145096
Step 261, loss: 0.07430049777030945
Step 262, loss: 0.0737222358584404
Step 263, loss: 0.07990670204162598
Step 264, loss: 0.07813651859760284
Step 265, loss: 0.08050532639026642
Step 266, loss: 0.07046013325452805
Step 267, loss: 0.08295343816280365
Step 268, loss: 0.0804486870765686
Step 269, loss: 0.07760566473007202
Step 270, loss: 0.07395690679550171
Step 271, loss: 0.07478272914886475
Step 272, loss: 0.0740082710981369
Step 273, loss: 0.08052319288253784
Step 274, loss: 0.07861707359552383
Step 275, loss: 0.070065438747406
Step 276, loss: 0.07930012792348862
Step 277, loss: 0.0766807347536087
Step 278, loss: 0.07297445833683014
Step 279, loss: 0.07423296570777893
Step 280, loss: 0.08846753090620041
Step 281, loss: 0.07492924481630325
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 282, loss: 0.07637274265289307
Step 283, loss: 0.07310236245393753
Step 284, loss: 0.08070990443229675
Step 285, loss: 0.0782109722495079
Step 286, loss: 0.07896459102630615
Step 287, loss: 0.0756256952881813
Step 288, loss: 0.08525692671537399
Step 289, loss: 0.07588425278663635
Step 290, loss: 0.07367654144763947
Step 291, loss: 0.07756517082452774
Step 292, loss: 0.07832346111536026
Step 293, loss: 0.07770068198442459
Step 294, loss: 0.07689868658781052
Step 295, loss: 0.0834919810295105
Step 296, loss: 0.07497106492519379
Step 297, loss: 0.08017727732658386
Step 298, loss: 0.07850449532270432
Step 299, loss: 0.07292384654283524
Step 300, loss: 0.07463440299034119
Step 301, loss: 0.07129716128110886
Step 302, loss: 0.06558706611394882
Step 303, loss: 0.07749452441930771
Step 304, loss: 0.0751955583691597
Step 305, loss: 0.08088627457618713
Step 306, loss: 0.0696171447634697
Step 307, loss: 0.07574302703142166
Step 308, loss: 0.07219752669334412
Step 309, loss: 0.0674944669008255
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 310, loss: 0.08049409091472626
Step 311, loss: 0.07577070593833923
Step 312, loss: 0.07124656438827515
Step 313, loss: 0.07488822191953659
Step 314, loss: 0.07304789870977402
Step 315, loss: 0.07174596935510635
Step 316, loss: 0.07227737456560135
Step 317, loss: 0.07370444387197495
Step 318, loss: 0.07146668434143066
Step 319, loss: 0.0734555572271347
Step 320, loss: 0.07185731083154678
Step 321, loss: 0.07193884998559952
Step 322, loss: 0.07924008369445801
Step 323, loss: 0.07459127902984619
Step 324, loss: 0.07185470312833786
Step 325, loss: 0.06916928291320801
Step 326, loss: 0.07558414340019226
Step 327, loss: 0.06881807744503021
Step 328, loss: 0.06865181028842926
Step 329, loss: 0.06877921521663666
Step 330, loss: 0.08298385143280029
Step 331, loss: 0.07279621809720993
Step 332, loss: 0.07208257913589478
Step 333, loss: 0.06424395740032196
Step 334, loss: 0.06817857176065445
Step 335, loss: 0.0769742876291275
Step 336, loss: 0.07591510564088821
Step 337, loss: 0.06792476028203964
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 338, loss: 0.06976059824228287
Step 339, loss: 0.062386177480220795
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 340, loss: 0.0757562667131424
Step 341, loss: 0.07193725556135178
Step 342, loss: 0.07409036159515381
Step 343, loss: 0.07155895233154297
Step 344, loss: 0.07619118690490723
Step 345, loss: 0.06724642962217331
Step 346, loss: 0.0686996653676033
Step 347, loss: 0.06960780918598175
Step 348, loss: 0.07117796689271927
Step 349, loss: 0.07235004752874374
Step 350, loss: 0.07289395481348038
Step 351, loss: 0.07124904543161392
Step 352, loss: 0.07591135054826736
Step 353, loss: 0.05938150733709335
Step 354, loss: 0.07141871750354767
Step 355, loss: 0.06971196830272675
Step 356, loss: 0.06641649454832077
Step 357, loss: 0.07176797091960907
Step 358, loss: 0.06548664718866348
Step 359, loss: 0.06743791699409485
Step 360, loss: 0.06443946063518524
Step 361, loss: 0.06936684995889664
Step 362, loss: 0.06579457223415375
Step 363, loss: 0.07219180464744568
Step 364, loss: 0.07256107032299042
Step 365, loss: 0.06914844363927841
Step 366, loss: 0.07107525318861008
Step 367, loss: 0.07048285752534866
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 368, loss: 0.06775183975696564
Step 369, loss: 0.07526548951864243
Step 370, loss: 0.06357400864362717
Step 371, loss: 0.07423649728298187
Step 372, loss: 0.06395532190799713
Step 373, loss: 0.06990110129117966
Step 374, loss: 0.06501936167478561
Step 375, loss: 0.06927185505628586
Step 376, loss: 0.07107315957546234
Step 377, loss: 0.0748443603515625
Step 378, loss: 0.07349420338869095
Step 379, loss: 0.06257882714271545
Step 380, loss: 0.0690067932009697
Step 381, loss: 0.0625036209821701
Step 382, loss: 0.06589680165052414
Step 383, loss: 0.06740155071020126
Step 384, loss: 0.07140204310417175
Step 385, loss: 0.06235720217227936
Step 386, loss: 0.06558951735496521
Step 387, loss: 0.0689176470041275
Step 388, loss: 0.06396230310201645
Step 389, loss: 0.06755638867616653
Step 390, loss: 0.07469333708286285
Step 391, loss: 0.07333772629499435
Step 392, loss: 0.07403210550546646
Step 393, loss: 0.06757253408432007
Step 394, loss: 0.07061048597097397
Step 395, loss: 0.07517968118190765
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 454, loss: 0.06368210911750793
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 396, loss: 0.06625440716743469
Step 397, loss: 0.07073629647493362
Step 398, loss: 0.06995418667793274
Step 399, loss: 0.06800923496484756
Step 400, loss: 0.06958360224962234
Step 401, loss: 0.0732586681842804
Step 402, loss: 0.06934764236211777
Step 403, loss: 0.06613286584615707
Step 404, loss: 0.06818868219852448
Step 405, loss: 0.06271519511938095
Step 406, loss: 0.0641980916261673
Step 407, loss: 0.06857927143573761
Step 408, loss: 0.06678001582622528
Step 409, loss: 0.06687375903129578
Step 410, loss: 0.06939789652824402
Step 411, loss: 0.06887051463127136
Step 412, loss: 0.06841285526752472
Step 413, loss: 0.06486549228429794
Step 414, loss: 0.06286819279193878
Step 415, loss: 0.06147408485412598
Step 416, loss: 0.0670069083571434
Step 417, loss: 0.06688813865184784
Step 418, loss: 0.06760692596435547
Step 419, loss: 0.06752729415893555
Step 420, loss: 0.06255713105201721
Step 421, loss: 0.05930148437619209
Step 422, loss: 0.060488924384117126
Step 423, loss: 0.06407476216554642
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 454, loss: 0.06368210911750793
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 424, loss: 0.0693524181842804
Step 425, loss: 0.07133162021636963
Step 426, loss: 0.060460228472948074
Step 427, loss: 0.07239401340484619
Step 428, loss: 0.06929553300142288
Step 429, loss: 0.06927846372127533
Step 430, loss: 0.06296417117118835
Step 431, loss: 0.06554847210645676
Step 432, loss: 0.0665377676486969
Step 433, loss: 0.07277698814868927
Step 434, loss: 0.06467586755752563
Step 435, loss: 0.07210275530815125
Step 436, loss: 0.06937238574028015
Step 437, loss: 0.0664115622639656
Step 438, loss: 0.06228533014655113
Step 439, loss: 0.06611409038305283
Step 440, loss: 0.06622735410928726
Step 441, loss: 0.07378384470939636
Step 442, loss: 0.06737452745437622
Step 443, loss: 0.06048573553562164
Step 444, loss: 0.06945990771055222
Step 445, loss: 0.06379619985818863
Step 446, loss: 0.062341559678316116
Step 447, loss: 0.06410534679889679
Step 448, loss: 0.06199028342962265
Step 449, loss: 0.06241285055875778
Step 450, loss: 0.06345609575510025
Step 451, loss: 0.062296438962221146
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 452, loss: 0.06334012746810913
Step 453, loss: 0.06873036921024323
Step 454, loss: 0.06368210911750793
Step 454, loss: 0.06368210911750793
Step 454, loss: 0.06368210911750793
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 567, loss: 0.05724496394395828
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 567, loss: 0.05724496394395828
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 567, loss: 0.05724496394395828
Step 567, loss: 0.05724496394395828
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 567, loss: 0.05724496394395828
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 567, loss: 0.05724496394395828
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 567, loss: 0.05724496394395828
Step 567, loss: 0.05724496394395828
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 567, loss: 0.05724496394395828
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 567, loss: 0.05724496394395828
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 567, loss: 0.05724496394395828
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 567, loss: 0.05724496394395828
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 567, loss: 0.05724496394395828
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 455, loss: 0.06426595151424408
Step 456, loss: 0.06304182857275009
Step 457, loss: 0.058793991804122925
Step 458, loss: 0.06457367539405823
Step 459, loss: 0.060991786420345306
Step 460, loss: 0.06375886499881744
Step 461, loss: 0.06415148079395294
Step 462, loss: 0.060376085340976715
Step 463, loss: 0.06056772917509079
Step 464, loss: 0.06895369291305542
Step 465, loss: 0.0671730786561966
Step 466, loss: 0.06665585190057755
Step 467, loss: 0.06260743737220764
Step 468, loss: 0.060622505843639374
Step 469, loss: 0.06348726153373718
Step 470, loss: 0.06622584909200668
Step 471, loss: 0.06089667230844498
Step 472, loss: 0.06946885585784912
Step 473, loss: 0.060297220945358276
Step 474, loss: 0.0626857727766037
Step 475, loss: 0.062131330370903015
Step 476, loss: 0.06241990625858307
Step 477, loss: 0.06374245136976242
Step 478, loss: 0.06332319229841232
Step 479, loss: 0.062388211488723755
Step 480, loss: 0.06365986168384552
Step 481, loss: 0.05709860473871231
Step 482, loss: 0.06301204115152359
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 483, loss: 0.061044931411743164
Step 484, loss: 0.05907278135418892
Step 485, loss: 0.05971646308898926
Step 486, loss: 0.06447527557611465
Step 487, loss: 0.05975550785660744
Step 488, loss: 0.06044264882802963
Step 489, loss: 0.060211580246686935
Step 490, loss: 0.06209130585193634
Step 491, loss: 0.06543129682540894
Step 492, loss: 0.059995707124471664
Step 493, loss: 0.06253092736005783
Step 494, loss: 0.06457742303609848
Step 495, loss: 0.060628000646829605
Step 496, loss: 0.062237273901700974
Step 497, loss: 0.059879668056964874
Step 498, loss: 0.06087687984108925
Step 499, loss: 0.06576115638017654
Step 500, loss: 0.061334457248449326
Step 501, loss: 0.055974457412958145
Step 502, loss: 0.061235811561346054
Step 503, loss: 0.06285053491592407
Step 504, loss: 0.06015505641698837
Step 505, loss: 0.053896550089120865
Step 506, loss: 0.05612781271338463
Step 507, loss: 0.05887729302048683
Step 508, loss: 0.05585925653576851
Step 509, loss: 0.0597778856754303
Step 510, loss: 0.057181838899850845
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 567, loss: 0.05724496394395828
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 511, loss: 0.061694033443927765
Step 512, loss: 0.06407112628221512
Step 513, loss: 0.05810495465993881
Step 514, loss: 0.06638425588607788
Step 515, loss: 0.06234056502580643
Step 516, loss: 0.05726346746087074
Step 517, loss: 0.055567625910043716
Step 518, loss: 0.055955637246370316
Step 519, loss: 0.05651025474071503
Step 520, loss: 0.05880095064640045
Step 521, loss: 0.06194344907999039
Step 522, loss: 0.054551199078559875
Step 523, loss: 0.05326776206493378
Step 524, loss: 0.06028522551059723
Step 525, loss: 0.05947022885084152
Step 526, loss: 0.0640629231929779
Step 527, loss: 0.05773034319281578
Step 528, loss: 0.06106280907988548
Step 529, loss: 0.06265497952699661
Step 530, loss: 0.0635722354054451
Step 531, loss: 0.05912148207426071
Step 532, loss: 0.05689976364374161
Step 533, loss: 0.05771622061729431
Step 534, loss: 0.05746182054281235
Step 535, loss: 0.05416140332818031
Step 536, loss: 0.057220157235860825
Step 537, loss: 0.058656152337789536
Step 538, loss: 0.05574904382228851
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 539, loss: 0.05597524344921112
Step 540, loss: 0.05773285776376724
Step 541, loss: 0.05242231488227844
Step 542, loss: 0.05662727355957031
Step 543, loss: 0.05508505553007126
Step 544, loss: 0.05941224843263626
Step 545, loss: 0.05821644514799118
Step 546, loss: 0.057495053857564926
Step 547, loss: 0.057394012808799744
Step 548, loss: 0.056536756455898285
Step 549, loss: 0.060356639325618744
Step 550, loss: 0.04997852072119713
Step 551, loss: 0.05949516221880913
Step 552, loss: 0.05548206344246864
Step 553, loss: 0.05847559869289398
Step 554, loss: 0.05835075303912163
Step 555, loss: 0.05459621548652649
Step 556, loss: 0.056156594306230545
Step 557, loss: 0.06219186261296272
Step 558, loss: 0.05685392767190933
Step 559, loss: 0.058768440037965775
Step 560, loss: 0.05872751772403717
Step 561, loss: 0.05837298557162285
Step 562, loss: 0.057191770523786545
Step 563, loss: 0.06310658156871796
Step 564, loss: 0.05979342386126518
Step 565, loss: 0.062125254422426224
Step 566, loss: 0.05740165710449219
Step 567, loss: 0.05724496394395828
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 567, loss: 0.05724496394395828
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 568, loss: 0.060693636536598206
Step 569, loss: 0.05657193064689636
Step 570, loss: 0.057977866381406784
Step 571, loss: 0.061129432171583176
Step 572, loss: 0.06126067787408829
Step 573, loss: 0.06667599827051163
Step 574, loss: 0.061891790479421616
Step 575, loss: 0.0659869983792305
Step 576, loss: 0.05749592185020447
Step 577, loss: 0.05431164801120758
Step 578, loss: 0.06304016709327698
Step 579, loss: 0.061839666217565536
Step 580, loss: 0.05399380624294281
Step 581, loss: 0.053802818059921265
Step 582, loss: 0.054317597299814224
Step 583, loss: 0.05783277377486229
Step 584, loss: 0.06007624790072441
Step 585, loss: 0.05594527721405029
Step 586, loss: 0.05605454742908478
Step 587, loss: 0.05816033110022545
Step 588, loss: 0.05568436160683632
Step 589, loss: 0.05114887282252312
Step 590, loss: 0.05531812459230423
Step 591, loss: 0.062178269028663635
Step 592, loss: 0.058294814079999924
Step 593, loss: 0.05578942596912384
Step 594, loss: 0.061778198927640915
Step 595, loss: 0.05845404416322708
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 596, loss: 0.05749756842851639
Step 597, loss: 0.05922927334904671
Step 598, loss: 0.05816197767853737
Step 599, loss: 0.06352390348911285
Step 600, loss: 0.05076915770769119
Step 601, loss: 0.05631988123059273
Step 602, loss: 0.05564216151833534
Step 603, loss: 0.05661308020353317
Step 604, loss: 0.05975494161248207
Step 605, loss: 0.05573396757245064
Step 606, loss: 0.05331631004810333
Step 607, loss: 0.05293450504541397
Step 608, loss: 0.05792666971683502
Step 609, loss: 0.05730050429701805
Step 610, loss: 0.05770248919725418
Step 611, loss: 0.05762555077672005
Step 612, loss: 0.05454135313630104
Step 613, loss: 0.058757707476615906
Step 614, loss: 0.054648514837026596
Step 615, loss: 0.05632779374718666
Step 616, loss: 0.05490336939692497
Step 617, loss: 0.053446732461452484
Step 618, loss: 0.06077239289879799
Step 619, loss: 0.05296066030859947
Step 620, loss: 0.05429023876786232
Step 621, loss: 0.05421148240566254
Step 622, loss: 0.05607752501964569
Step 623, loss: 0.055563945323228836
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 624, loss: 0.05999525263905525
Step 625, loss: 0.05581849068403244
Step 626, loss: 0.05863697826862335
Step 627, loss: 0.05963205546140671
Step 628, loss: 0.054530952125787735
Step 629, loss: 0.059330206364393234
Step 630, loss: 0.058975670486688614
Step 631, loss: 0.05490532144904137
Step 632, loss: 0.05183026194572449
Step 633, loss: 0.05531543493270874
Step 634, loss: 0.051033347845077515
Step 635, loss: 0.052353598177433014
Step 636, loss: 0.054740991443395615
Step 637, loss: 0.054972998797893524
Step 638, loss: 0.055963002145290375
Step 639, loss: 0.05635948106646538
Step 640, loss: 0.053303062915802
Step 641, loss: 0.05401785671710968
Step 642, loss: 0.05358312651515007
Step 643, loss: 0.05682080239057541
Step 644, loss: 0.060251157730817795
Step 645, loss: 0.05708666890859604
Step 646, loss: 0.057269543409347534
Step 647, loss: 0.05856398865580559
Step 648, loss: 0.058661773800849915
Step 649, loss: 0.06284508109092712
Step 650, loss: 0.05223540589213371
Step 651, loss: 0.06013790890574455
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 652, loss: 0.058416496962308884
Step 653, loss: 0.05345001444220543
Step 654, loss: 0.0571986585855484
Step 655, loss: 0.05246962606906891
Step 656, loss: 0.054803039878606796
Step 657, loss: 0.05862417444586754
Step 658, loss: 0.055395953357219696
Step 659, loss: 0.051739275455474854
Step 660, loss: 0.05647405982017517
Step 661, loss: 0.05281607434153557
Step 662, loss: 0.054644983261823654
Step 663, loss: 0.052483316510915756
Step 664, loss: 0.05117356777191162
Step 665, loss: 0.056670818477869034
Step 666, loss: 0.0524410679936409
Step 667, loss: 0.05108580365777016
Step 668, loss: 0.053567394614219666
Step 669, loss: 0.054822467267513275
Step 670, loss: 0.05470137298107147
Step 671, loss: 0.057231348007917404
Step 672, loss: 0.05458701401948929
Step 673, loss: 0.05213673412799835
Step 674, loss: 0.05331171303987503
Step 675, loss: 0.05683687701821327
Step 676, loss: 0.052388016134500504
Step 677, loss: 0.05773136392235756
Step 678, loss: 0.05292825028300285
Step 679, loss: 0.054960232228040695
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 792, loss: 0.05276314914226532
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 792, loss: 0.05276314914226532
Step 792, loss: 0.05276314914226532
Step 792, loss: 0.05276314914226532
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 905, loss: 0.046647969633340836
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 905, loss: 0.046647969633340836
Step 905, loss: 0.046647969633340836
Step 905, loss: 0.046647969633340836
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 792, loss: 0.05276314914226532
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 792, loss: 0.05276314914226532
Step 792, loss: 0.05276314914226532
Step 792, loss: 0.05276314914226532
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 905, loss: 0.046647969633340836
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 905, loss: 0.046647969633340836
Step 792, loss: 0.05276314914226532
Step 792, loss: 0.05276314914226532
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 792, loss: 0.05276314914226532
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 792, loss: 0.05276314914226532
Step 792, loss: 0.05276314914226532
Step 905, loss: 0.046647969633340836
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 905, loss: 0.046647969633340836
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 905, loss: 0.046647969633340836
Step 905, loss: 0.046647969633340836
Step 905, loss: 0.046647969633340836
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 680, loss: 0.06042022630572319
Step 681, loss: 0.05161309987306595
Step 682, loss: 0.05512462556362152
Step 683, loss: 0.05846741795539856
Step 684, loss: 0.04999315366148949
Step 685, loss: 0.05427156016230583
Step 686, loss: 0.052401572465896606
Step 687, loss: 0.05806005746126175
Step 688, loss: 0.058303747326135635
Step 689, loss: 0.05887821689248085
Step 690, loss: 0.05506887659430504
Step 691, loss: 0.05558210238814354
Step 692, loss: 0.05234615504741669
Step 693, loss: 0.05609278380870819
Step 694, loss: 0.05400462821125984
Step 695, loss: 0.051567479968070984
Step 696, loss: 0.0525931790471077
Step 697, loss: 0.05266290158033371
Step 698, loss: 0.05522635951638222
Step 699, loss: 0.057368505746126175
Step 700, loss: 0.055467113852500916
Step 701, loss: 0.05440811812877655
Step 702, loss: 0.057939570397138596
Step 703, loss: 0.060186367481946945
Step 704, loss: 0.05358351022005081
Step 705, loss: 0.04909184202551842
Step 706, loss: 0.0541643463075161
Step 707, loss: 0.05551610514521599
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 708, loss: 0.052318211644887924
Step 709, loss: 0.05198443681001663
Step 710, loss: 0.0531727634370327
Step 711, loss: 0.050181228667497635
Step 712, loss: 0.05198992043733597
Step 713, loss: 0.055155761539936066
Step 714, loss: 0.053536444902420044
Step 715, loss: 0.05513673275709152
Step 716, loss: 0.051219530403614044
Step 717, loss: 0.050449274480342865
Step 718, loss: 0.053337596356868744
Step 719, loss: 0.050231270492076874
Step 720, loss: 0.05812658742070198
Step 721, loss: 0.05324537307024002
Step 722, loss: 0.05423666909337044
Step 723, loss: 0.04970180615782738
Step 724, loss: 0.05189107730984688
Step 725, loss: 0.056188758462667465
Step 726, loss: 0.0578262060880661
Step 727, loss: 0.05336611345410347
Step 728, loss: 0.054934319108724594
Step 729, loss: 0.054610975086688995
Step 730, loss: 0.054616864770650864
Step 731, loss: 0.05188506096601486
Step 732, loss: 0.05122481286525726
Step 733, loss: 0.054014693945646286
Step 734, loss: 0.05114324763417244
Step 735, loss: 0.05204615741968155
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 736, loss: 0.049834612756967545
Step 737, loss: 0.05334187299013138
Step 738, loss: 0.05381673574447632
Step 739, loss: 0.05684928968548775
Step 740, loss: 0.05198483541607857
Step 741, loss: 0.05248570814728737
Step 742, loss: 0.053286731243133545
Step 743, loss: 0.047883909195661545
Step 744, loss: 0.05513735115528107
Step 745, loss: 0.04962250590324402
Step 746, loss: 0.055173471570014954
Step 747, loss: 0.05802896246314049
Step 748, loss: 0.05428009480237961
Step 749, loss: 0.05088415741920471
Step 750, loss: 0.05142022669315338
Step 751, loss: 0.05234065651893616
Step 752, loss: 0.05555622652173042
Step 753, loss: 0.05374646931886673
Step 754, loss: 0.05180392786860466
Step 755, loss: 0.05009395256638527
Step 756, loss: 0.05198207497596741
Step 757, loss: 0.05238225311040878
Step 758, loss: 0.05293664708733559
Step 759, loss: 0.05449116230010986
Step 760, loss: 0.04926205426454544
Step 761, loss: 0.053567323833703995
Step 762, loss: 0.04813891649246216
Step 763, loss: 0.05183335766196251
Step 792, loss: 0.05276314914226532
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 764, loss: 0.048774633556604385
Step 765, loss: 0.05351027473807335
Step 766, loss: 0.05433560535311699
Step 767, loss: 0.05625997856259346
Step 768, loss: 0.056353695690631866
Step 769, loss: 0.0508299246430397
Step 770, loss: 0.05155382305383682
Step 771, loss: 0.049010269343853
Step 772, loss: 0.05568309128284454
Step 773, loss: 0.05092721804976463
Step 774, loss: 0.05019589886069298
Step 775, loss: 0.05267471820116043
Step 776, loss: 0.05427205190062523
Step 777, loss: 0.05368969589471817
Step 778, loss: 0.05210775509476662
Step 779, loss: 0.054233841598033905
Step 780, loss: 0.05037105828523636
Step 781, loss: 0.05182868242263794
Step 782, loss: 0.054920949041843414
Step 783, loss: 0.047672830522060394
Step 784, loss: 0.04955824464559555
Step 785, loss: 0.054256193339824677
Step 786, loss: 0.054079193621873856
Step 787, loss: 0.05246082320809364
Step 788, loss: 0.05066518485546112
Step 789, loss: 0.053069982677698135
Step 790, loss: 0.057384900748729706
Step 791, loss: 0.04665083438158035
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 792, loss: 0.05276314914226532
Step 905, loss: 0.046647969633340836
Step 792, loss: 0.05276314914226532
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 793, loss: 0.054061852395534515
Step 794, loss: 0.048088300973176956
Step 795, loss: 0.04994717985391617
Step 796, loss: 0.050434160977602005
Step 797, loss: 0.049374524503946304
Step 798, loss: 0.05027833953499794
Step 799, loss: 0.05122183635830879
Step 800, loss: 0.0489298515021801
Step 801, loss: 0.05107594281435013
Step 802, loss: 0.051183171570301056
Step 803, loss: 0.055726561695337296
Step 804, loss: 0.05236414074897766
Step 805, loss: 0.05481838062405586
Step 806, loss: 0.05260610580444336
Step 807, loss: 0.0543585866689682
Step 808, loss: 0.05203908681869507
Step 809, loss: 0.05462546646595001
Step 810, loss: 0.053835030645132065
Step 811, loss: 0.05216866731643677
Step 812, loss: 0.05362238734960556
Step 813, loss: 0.0498078316450119
Step 814, loss: 0.05364372581243515
Step 815, loss: 0.05224821716547012
Step 816, loss: 0.04948127642273903
Step 817, loss: 0.04813219606876373
Step 818, loss: 0.05043148621916771
Step 819, loss: 0.05536743253469467
Step 820, loss: 0.05353039130568504
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 821, loss: 0.0544835664331913
Step 822, loss: 0.0528404600918293
Step 823, loss: 0.05409259349107742
Step 824, loss: 0.05190572142601013
Step 825, loss: 0.05433269217610359
Step 826, loss: 0.05264841765165329
Step 827, loss: 0.05401146411895752
Step 828, loss: 0.04975510016083717
Step 829, loss: 0.0521039217710495
Step 830, loss: 0.0499560609459877
Step 831, loss: 0.05116429552435875
Step 832, loss: 0.048538923263549805
Step 833, loss: 0.04904584214091301
Step 834, loss: 0.04836011305451393
Step 835, loss: 0.05034517124295235
Step 836, loss: 0.05005499720573425
Step 837, loss: 0.05007874220609665
Step 838, loss: 0.047859881073236465
Step 839, loss: 0.04532647132873535
Step 840, loss: 0.055694304406642914
Step 841, loss: 0.0554664246737957
Step 842, loss: 0.05067469924688339
Step 843, loss: 0.049735430628061295
Step 844, loss: 0.05013083294034004
Step 845, loss: 0.05104135349392891
Step 846, loss: 0.048111334443092346
Step 847, loss: 0.048332344740629196
Step 848, loss: 0.05154590681195259
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 849, loss: 0.05339287593960762
Step 850, loss: 0.056519683450460434
Step 851, loss: 0.05075613781809807
Step 852, loss: 0.0623568594455719
Step 853, loss: 0.0549653060734272
Step 854, loss: 0.05397307127714157
Step 855, loss: 0.04982191324234009
Step 856, loss: 0.048964910209178925
Step 857, loss: 0.047429073601961136
Step 858, loss: 0.04729593172669411
Step 859, loss: 0.05210305005311966
Step 860, loss: 0.052235037088394165
Step 861, loss: 0.0512537956237793
Step 862, loss: 0.05365293100476265
Step 863, loss: 0.04893507435917854
Step 864, loss: 0.049814388155937195
Step 865, loss: 0.05050407350063324
Step 866, loss: 0.0510561428964138
Step 867, loss: 0.05000585317611694
Step 868, loss: 0.05183959752321243
Step 869, loss: 0.050178587436676025
Step 870, loss: 0.05730721354484558
Step 871, loss: 0.047492437064647675
Step 872, loss: 0.05254639685153961
Step 873, loss: 0.05025511980056763
Step 874, loss: 0.052888110280036926
Step 875, loss: 0.05028617009520531
Step 876, loss: 0.050423216074705124
Step 905, loss: 0.046647969633340836
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 877, loss: 0.05188605189323425
Step 878, loss: 0.047956787049770355
Step 879, loss: 0.04578006640076637
Step 880, loss: 0.04656338691711426
Step 881, loss: 0.05148216709494591
Step 882, loss: 0.050687819719314575
Step 883, loss: 0.04962923750281334
Step 884, loss: 0.04921175166964531
Step 885, loss: 0.04796570912003517
Step 886, loss: 0.049120448529720306
Step 887, loss: 0.051558949053287506
Step 888, loss: 0.05483433976769447
Step 889, loss: 0.04656941816210747
Step 890, loss: 0.04760434478521347
Step 891, loss: 0.05015144869685173
Step 892, loss: 0.05053212493658066
Step 893, loss: 0.050063204020261765
Step 894, loss: 0.04825127124786377
Step 895, loss: 0.05196908861398697
Step 896, loss: 0.050996001809835434
Step 897, loss: 0.050228703767061234
Step 898, loss: 0.050335243344306946
Step 899, loss: 0.046443093568086624
Step 900, loss: 0.05043061450123787
Step 901, loss: 0.04934031143784523
Step 902, loss: 0.05054517090320587
Step 903, loss: 0.049222659319639206
Step 904, loss: 0.0462678000330925
Step 905, loss: 0.046647969633340836
Step 905, loss: 0.046647969633340836
Step 905, loss: 0.046647969633340836
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    await asyncio.sleep(0.25)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, 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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
Traceback (most recent call last):
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    await asyncio.sleep(0.25)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    await asyncio.sleep(0.25)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    await asyncio.sleep(0.25)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    return await future
    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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, 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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    return fut.result()
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    return fut.result()
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    return future.result()
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    await asyncio.gather(*write_coros)
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    raise ValueError(
ValueError: [process_index=12] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    await asyncio.gather(*write_coros)
    raise ValueError(
ValueError: [process_index=13] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    await self._maybe_create_base_dir(file_path.parent)
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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=14] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    raise ValueError(
    raise ValueError(
ValueError: [process_index=15] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
ValueError: [process_index=8] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679466_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    raise ValueError(
ValueError: [process_index=11] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679466_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/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=10] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679466_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679466_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    await asyncio.sleep(0.25)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
Traceback (most recent call last):
Step 906, loss: 0.04930618405342102
Step 907, loss: 0.04940510913729668
Step 908, loss: 0.04552031308412552
Step 909, loss: 0.05029265210032463
Step 910, loss: 0.047847453504800797
Step 911, loss: 0.04745199531316757
Step 912, loss: 0.05232144892215729
Step 913, loss: 0.04972940683364868
Step 914, loss: 0.04501498490571976
Step 915, loss: 0.04650250822305679
Step 916, loss: 0.0495048388838768
Step 917, loss: 0.04650386422872543
Step 918, loss: 0.05281542241573334
Step 919, loss: 0.04914632439613342
Step 920, loss: 0.04543978348374367
Step 921, loss: 0.04979666322469711
Step 922, loss: 0.04555828124284744
Step 923, loss: 0.045400697737932205
Step 924, loss: 0.04834138974547386
Step 925, loss: 0.04368482157588005
Step 926, loss: 0.04732787236571312
Step 927, loss: 0.04456918314099312
Step 928, loss: 0.044946204870939255
Step 929, loss: 0.04813944548368454
Step 930, loss: 0.04508523643016815
Step 931, loss: 0.04880047217011452
Step 932, loss: 0.04543484002351761
Step 933, loss: 0.045454464852809906
Step 934, loss: 0.04879076033830643
Step 935, loss: 0.043344106525182724
Step 936, loss: 0.04851078242063522
Step 937, loss: 0.04931316152215004
Step 938, loss: 0.04784514382481575
Step 939, loss: 0.04343545436859131
Step 940, loss: 0.04468730092048645
Step 941, loss: 0.04562007635831833
Step 942, loss: 0.0513860322535038
Step 943, loss: 0.0457218736410141
Step 944, loss: 0.04683634638786316
Step 945, loss: 0.049558911472558975
Step 946, loss: 0.050162650644779205
Step 947, loss: 0.04794125631451607
Step 948, loss: 0.050296761095523834
Step 949, loss: 0.0434456430375576
Step 950, loss: 0.04430101439356804
Step 951, loss: 0.044370055198669434
Step 952, loss: 0.048227764666080475
Step 953, loss: 0.04192202538251877
Step 954, loss: 0.04803616926074028
Step 955, loss: 0.04753837734460831
Step 956, loss: 0.046032555401325226
Step 957, loss: 0.04530402645468712
Step 958, loss: 0.04444444924592972
Step 959, loss: 0.042897067964076996
Step 960, loss: 0.046944510191679
Step 961, loss: 0.045023590326309204
Step 962, loss: 0.04943760484457016
Step 963, loss: 0.0441926009953022
Step 964, loss: 0.04464635252952576
Step 965, loss: 0.0428805835545063
Step 966, loss: 0.041962359100580215
Step 967, loss: 0.044932760298252106
Step 968, loss: 0.04574669525027275
Step 969, loss: 0.04803662747144699
Step 970, loss: 0.04424861818552017
Step 971, loss: 0.044142719358205795
Step 972, loss: 0.043939366936683655
Step 973, loss: 0.04521915689110756
Step 974, loss: 0.04488588124513626
Step 975, loss: 0.04055127501487732
Step 976, loss: 0.045942701399326324
Step 977, loss: 0.044177886098623276
Step 978, loss: 0.04540722444653511
Step 979, loss: 0.049032412469387054
Step 980, loss: 0.04757321625947952
Step 981, loss: 0.050052616745233536
Step 982, loss: 0.04291177913546562
Step 983, loss: 0.04917789623141289
Step 984, loss: 0.04504353180527687
Step 985, loss: 0.0483022965490818
Step 986, loss: 0.04472198709845543
Step 987, loss: 0.04395439848303795
Step 988, loss: 0.04285021126270294
Step 989, loss: 0.04739050567150116
Step 990, loss: 0.04370202496647835
Step 991, loss: 0.04467702656984329
Step 992, loss: 0.04954266920685768
Step 993, loss: 0.04399842023849487
Step 994, loss: 0.041520923376083374
Step 995, loss: 0.04435744136571884
Step 996, loss: 0.04402275010943413
Step 997, loss: 0.045924365520477295
Step 998, loss: 0.04502137005329132
Step 999, loss: 0.04693201929330826
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 159, in wait_for_base_dir_creation
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 605, in sleep
    await asyncio.sleep(0.25)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
    return await future
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    return await future
asyncio.exceptions.CancelledError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 456, in wait_for
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, 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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 162, in _maybe_create_base_dir
    raise exceptions.TimeoutError() from exc
asyncio.exceptions.TimeoutError
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for

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

Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    await asyncio.wait_for(
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/tasks.py", line 458, in wait_for
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py", line 255, in save
    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-p0023960/tum_cte0515/Projects/jafar_jobs/train_tokenizer.py", line 273, in <module>
    orbax_checkpointer.save(
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py", line 594, in save
    self._handler.save(tmpdir.get(), args=ckpt_args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 554, in save
    self._handler_impl.save(directory, args=args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    asyncio_utils.run_sync(async_save(directory, *args, **kwargs))
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py", line 552, in async_save
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 174, in result
    f.result()  # Block on result.
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 397, in result
    f.result(timeout=time_remaining)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    return self._f.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 348, in result
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    return self._t.result(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 297, in result
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/threading.py", line 953, in run
    self.join(timeout=timeout)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 294, in join
    raise self._exception
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 281, in run
    super().run()
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
    self._target(*self._args, **self._kwargs)
    self._target(*self._args, **self._kwargs)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 230, in _target_setting_result
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/futures/future.py", line 341, in <lambda>
    self._result = target()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    target=lambda: asyncio_utils.run_sync(coro),
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/asyncio_utils.py", line 50, in run_sync
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return future.result()
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py", line 1105, in _background_serialize
    return asyncio.run(coro)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/home/hk-project-p0023960/tum_cte0515/.local/share/uv/python/cpython-3.10.18-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-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 194, in write
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    await asyncio.gather(*write_coros)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py", line 169, in _maybe_create_base_dir
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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=6] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    await self._maybe_create_base_dir(file_path.parent)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/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=4] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    raise ValueError(
ValueError: [process_index=7] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
    raise ValueError(
ValueError: [process_index=5] Timed out waiting for array_metadatas base directory creation: /mnt/odfs/3318557/stripe_default/checkpoints/tokenizer_1751679472_1000.orbax-checkpoint-tmp-0/array_metadatas. timeout=600 seconds. primary_process=0
2025-07-05 04:28:26.912117: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']
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-05 04:28:26.912152: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError=''] [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.912457: 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-05 04:28:26.912595: 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-05 04:28:26.912626: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.912713: 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-05 04:28:26.912792: 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-05 04:28:26.912735: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.912763: 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-05 04:28:26.912726: 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-05 04:28:26.912806: 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-05 04:28:26.912727: 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-05 04:28:26.913180: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

2025-07-05 04:28:26.912648: 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-05 04:28:26.913345: 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-05 04:28:26.912756: 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-05 04:28:26.912674: 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-05 04:28:26.912705: 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-05 04:28:26.913185: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

2025-07-05 04:28:26.913104: 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-05 04:28:26.913070: 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-05 04:28:26.913181: 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-05 04:28:26.926675: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.926639: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.926738: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.926699: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.929922: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.929819: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.929966: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.929974: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.930870: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.930986: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.930870: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
2025-07-05 04:28:26.930997: 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::16040846885349643490::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: 12/16.
The first task at the barrier: /job:jax_worker/replica:0/task:9. Some timed out task names:
/job:jax_worker/replica:0/task:1
/job:jax_worker/replica:0/task:0
 [type.googleapis.com/tensorflow.BarrierError='
\x1eShutdown::16040846885349643490'] [type.googleapis.com/tensorflow.CoordinationServiceError='']

RPC: /tensorflow.CoordinationService/PollForError [type.googleapis.com/tensorflow.CoordinationServiceError='']
srun: error: hkn0511: task 11: Aborted (core dumped)
srun: error: hkn0519: task 15: Aborted (core dumped)
srun: error: hkn0414: tasks 0,2-3: Aborted (core dumped)
srun: error: hkn0511: tasks 9-10: Aborted (core dumped)
srun: error: hkn0416: tasks 4,6-7: Aborted (core dumped)
srun: error: hkn0519: tasks 13-14: Aborted (core dumped)
srun: error: hkn0511: task 8: Aborted (core dumped)
srun: error: hkn0416: task 5: Aborted (core dumped)
srun: error: hkn0519: task 12: Aborted (core dumped)
srun: error: hkn0414: task 1: Aborted (core dumped)

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

Job ID: 3318557
Cluster: hk
User/Group: tum_cte0515/hk-project-p0023960
Account: hk-project-p0023960
State: FAILED (exit code 134)
Partition: accelerated
Nodes: 4
Cores per node: 24
Nodelist: hkn[0414,0416,0511,0519]
CPU Utilized: 08:08:58
CPU Efficiency: 9.73% of 3-11:44:00 core-walltime
Job Wall-clock time: 00:52:20
Starttime: Sat Jul  5 03:36:09 2025
Endtime: Sat Jul  5 04:28:29 2025
Memory Utilized: 144.25 GB (estimated maximum)
Memory Efficiency: 0.00% of 0.00 MB (0.00 MB/node)
Energy Consumed: 15234470 Joule / 4231.79722222222 Watthours
Average node power draw: 4851.74203821656 Watt