diff --git "a/checkpoints/DiffCSP/Carbon24/result.out" "b/checkpoints/DiffCSP/Carbon24/result.out" new file mode 100644--- /dev/null +++ "b/checkpoints/DiffCSP/Carbon24/result.out" @@ -0,0 +1,3322 @@ +--dataset=Carbon24 --lr=5e-4 --lr_scheduler=None --epochs=1000 --num_evals=1 --batch_size=1024 --emb_dim=128 --output_model_dir=../output_DiffCSP/Carbon24/5e-4_None_1024_128_1_1000 +Wed 07 Feb 2024 12:18:25 PM EST +start +/gpfs/home/admin.shengchao.liu/Crystal_Generation_workspace/Crystallization_Generation_Diffusion_dev/crystallization/models/DiffCSP/common/data_utils.py:904: UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than torch.tensor(sourceTensor). + X = torch.tensor(X, dtype=torch.float) +model CSPDiffusion( + (decoder): CSPNet( + (node_embedding): Embedding(100, 512) + (atom_latent_emb): Linear(in_features=768, out_features=512, bias=True) + (act_fn): SiLU() + (dis_emb): SinusoidsEmbedding() + (csp_layer_0): CSPLayer( + (dis_emb): SinusoidsEmbedding() + (edge_mlp): Sequential( + (0): Linear(in_features=1801, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (node_mlp): Sequential( + (0): Linear(in_features=1024, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (csp_layer_1): CSPLayer( + (dis_emb): SinusoidsEmbedding() + (edge_mlp): Sequential( + (0): Linear(in_features=1801, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (node_mlp): Sequential( + (0): Linear(in_features=1024, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (csp_layer_2): CSPLayer( + (dis_emb): SinusoidsEmbedding() + (edge_mlp): Sequential( + (0): Linear(in_features=1801, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (node_mlp): Sequential( + (0): Linear(in_features=1024, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (csp_layer_3): CSPLayer( + (dis_emb): SinusoidsEmbedding() + (edge_mlp): Sequential( + (0): Linear(in_features=1801, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (node_mlp): Sequential( + (0): Linear(in_features=1024, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (csp_layer_4): CSPLayer( + (dis_emb): SinusoidsEmbedding() + (edge_mlp): Sequential( + (0): Linear(in_features=1801, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (node_mlp): Sequential( + (0): Linear(in_features=1024, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (csp_layer_5): CSPLayer( + (dis_emb): SinusoidsEmbedding() + (edge_mlp): Sequential( + (0): Linear(in_features=1801, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (node_mlp): Sequential( + (0): Linear(in_features=1024, out_features=512, bias=True) + (1): SiLU() + (2): Linear(in_features=512, out_features=512, bias=True) + (3): SiLU() + ) + (layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (coord_out): Linear(in_features=512, out_features=3, bias=False) + (lattice_out): Linear(in_features=512, out_features=9, bias=False) + (final_layer_norm): LayerNorm((512,), eps=1e-05, elementwise_affine=True) + ) + (beta_scheduler): BetaScheduler() + (sigma_scheduler): SigmaScheduler() + (time_embedding): SinusoidalTimeEmbeddings() +) + +epoch 1 +train lattice loss: 27.366275 coord loss: 1.212722 + +epoch 2 +train lattice loss: 3.745815 coord loss: 1.104478 + +epoch 3 +train lattice loss: 1.824620 coord loss: 1.065009 + +epoch 4 +train lattice loss: 1.190425 coord loss: 1.037542 + +epoch 5 +train lattice loss: 0.928119 coord loss: 1.018741 + +epoch 6 +train lattice loss: 0.848273 coord loss: 1.023920 + +epoch 7 +train lattice loss: 0.814765 coord loss: 1.006104 + +epoch 8 +train lattice loss: 0.774993 coord loss: 1.006602 + +epoch 9 +train lattice loss: 0.762115 coord loss: 1.001882 + +epoch 10 +train lattice loss: 0.737311 coord loss: 1.007151 +val lattice loss: 0.734063 coord loss: 1.005305 +save model with optimal loss + +epoch 11 +train lattice loss: 0.721234 coord loss: 1.002064 + +epoch 12 +train lattice loss: 0.719987 coord loss: 0.997250 + +epoch 13 +train lattice loss: 0.719670 coord loss: 1.005992 + +epoch 14 +train lattice loss: 0.713078 coord loss: 1.004345 + +epoch 15 +train lattice loss: 0.701057 coord loss: 1.004978 + +epoch 16 +train lattice loss: 0.701665 coord loss: 1.005150 + +epoch 17 +train lattice loss: 0.702574 coord loss: 1.000298 + +epoch 18 +train lattice loss: 0.691586 coord loss: 1.006122 + +epoch 19 +train lattice loss: 0.685900 coord loss: 1.000115 + +epoch 20 +train lattice loss: 0.681384 coord loss: 1.003991 +val lattice loss: 0.686480 coord loss: 0.993445 +save model with optimal loss + +epoch 21 +train lattice loss: 0.672676 coord loss: 1.002521 + +epoch 22 +train lattice loss: 0.673149 coord loss: 0.997800 + +epoch 23 +train lattice loss: 0.661546 coord loss: 1.000134 + +epoch 24 +train lattice loss: 0.664868 coord loss: 1.006283 + +epoch 25 +train lattice loss: 0.660338 coord loss: 0.999362 + +epoch 26 +train lattice loss: 0.652027 coord loss: 0.995094 + +epoch 27 +train lattice loss: 0.664279 coord loss: 1.002080 + +epoch 28 +train lattice loss: 0.659317 coord loss: 1.004260 + +epoch 29 +train lattice loss: 0.655371 coord loss: 1.002192 + +epoch 30 +train lattice loss: 0.649405 coord loss: 0.991293 +val lattice loss: 0.641754 coord loss: 0.994783 +save model with optimal loss + +epoch 31 +train lattice loss: 0.649504 coord loss: 1.003714 + +epoch 32 +train lattice loss: 0.639054 coord loss: 0.998804 + +epoch 33 +train lattice loss: 0.639887 coord loss: 1.001917 + +epoch 34 +train lattice loss: 0.651008 coord loss: 0.995854 + +epoch 35 +train lattice loss: 0.649804 coord loss: 0.998766 + +epoch 36 +train lattice loss: 0.636661 coord loss: 1.000541 + +epoch 37 +train lattice loss: 0.650482 coord loss: 0.996193 + +epoch 38 +train lattice loss: 0.641268 coord loss: 0.994108 + +epoch 39 +train lattice loss: 0.642590 coord loss: 0.993468 + +epoch 40 +train lattice loss: 0.630153 coord loss: 0.997944 +val lattice loss: 0.633550 coord loss: 0.998503 +save model with optimal loss + +epoch 41 +train lattice loss: 0.632670 coord loss: 0.996117 + +epoch 42 +train lattice loss: 0.623972 coord loss: 0.994346 + +epoch 43 +train lattice loss: 0.636762 coord loss: 0.994209 + +epoch 44 +train lattice loss: 0.634011 coord loss: 0.992135 + +epoch 45 +train lattice loss: 0.634555 coord loss: 0.996895 + +epoch 46 +train lattice loss: 0.627554 coord loss: 0.991847 + +epoch 47 +train lattice loss: 0.628312 coord loss: 0.993013 + +epoch 48 +train lattice loss: 0.625891 coord loss: 0.989980 + +epoch 49 +train lattice loss: 0.632308 coord loss: 0.987071 + +epoch 50 +train lattice loss: 0.627408 coord loss: 0.981846 +val lattice loss: 0.631549 coord loss: 0.980183 +save model with optimal loss + +epoch 51 +train lattice loss: 0.628666 coord loss: 0.985688 + +epoch 52 +train lattice loss: 0.607422 coord loss: 0.977652 + +epoch 53 +train lattice loss: 0.634566 coord loss: 0.983114 + +epoch 54 +train lattice loss: 0.613420 coord loss: 0.991000 + +epoch 55 +train lattice loss: 0.616798 coord loss: 0.985029 + +epoch 56 +train lattice loss: 0.610282 coord loss: 0.981177 + +epoch 57 +train lattice loss: 0.617293 coord loss: 0.980087 + +epoch 58 +train lattice loss: 0.618990 coord loss: 0.979058 + +epoch 59 +train lattice loss: 0.616362 coord loss: 0.983277 + +epoch 60 +train lattice loss: 0.638383 coord loss: 0.978895 +val lattice loss: 0.603186 coord loss: 0.977960 +save model with optimal loss + +epoch 61 +train lattice loss: 0.616890 coord loss: 0.984057 + +epoch 62 +train lattice loss: 0.622240 coord loss: 0.984512 + +epoch 63 +train lattice loss: 0.619370 coord loss: 0.972240 + +epoch 64 +train lattice loss: 0.612408 coord loss: 0.979354 + +epoch 65 +train lattice loss: 0.593136 coord loss: 0.978008 + +epoch 66 +train lattice loss: 0.604102 coord loss: 0.975055 + +epoch 67 +train lattice loss: 0.605334 coord loss: 0.980504 + +epoch 68 +train lattice loss: 0.614825 coord loss: 0.973260 + +epoch 69 +train lattice loss: 0.595758 coord loss: 0.972871 + +epoch 70 +train lattice loss: 0.595916 coord loss: 0.976083 +val lattice loss: 0.600418 coord loss: 0.971262 +save model with optimal loss + +epoch 71 +train lattice loss: 0.589001 coord loss: 0.970862 + +epoch 72 +train lattice loss: 0.595760 coord loss: 0.973390 + +epoch 73 +train lattice loss: 0.602594 coord loss: 0.973325 + +epoch 74 +train lattice loss: 0.602431 coord loss: 0.975329 + +epoch 75 +train lattice loss: 0.604988 coord loss: 0.971525 + +epoch 76 +train lattice loss: 0.610839 coord loss: 0.968766 + +epoch 77 +train lattice loss: 0.599273 coord loss: 0.979348 + +epoch 78 +train lattice loss: 0.601755 coord loss: 0.966048 + +epoch 79 +train lattice loss: 0.605946 coord loss: 0.973899 + +epoch 80 +train lattice loss: 0.589095 coord loss: 0.968516 +val lattice loss: 0.599044 coord loss: 0.971575 +save model with optimal loss + +epoch 81 +train lattice loss: 0.596854 coord loss: 0.971385 + +epoch 82 +train lattice loss: 0.591842 coord loss: 0.968366 + +epoch 83 +train lattice loss: 0.587170 coord loss: 0.967926 + +epoch 84 +train lattice loss: 0.574087 coord loss: 0.972368 + +epoch 85 +train lattice loss: 0.586804 coord loss: 0.967824 + +epoch 86 +train lattice loss: 0.600518 coord loss: 0.965657 + +epoch 87 +train lattice loss: 0.606965 coord loss: 0.968670 + +epoch 88 +train lattice loss: 0.603233 coord loss: 0.973692 + +epoch 89 +train lattice loss: 0.600380 coord loss: 0.966616 + +epoch 90 +train lattice loss: 0.590862 coord loss: 0.968949 +val lattice loss: 0.586839 coord loss: 0.969944 +save model with optimal loss + +epoch 91 +train lattice loss: 0.585878 coord loss: 0.971649 + +epoch 92 +train lattice loss: 0.593191 coord loss: 0.967873 + +epoch 93 +train lattice loss: 0.586758 coord loss: 0.964200 + +epoch 94 +train lattice loss: 0.592319 coord loss: 0.967674 + +epoch 95 +train lattice loss: 0.591934 coord loss: 0.973146 + +epoch 96 +train lattice loss: 0.584226 coord loss: 0.965802 + +epoch 97 +train lattice loss: 0.587459 coord loss: 0.970272 + +epoch 98 +train lattice loss: 0.580728 coord loss: 0.967071 + +epoch 99 +train lattice loss: 0.577153 coord loss: 0.965460 + +epoch 100 +train lattice loss: 0.582871 coord loss: 0.966271 +val lattice loss: 0.583756 coord loss: 0.962569 +save model with optimal loss + +epoch 101 +train lattice loss: 0.572796 coord loss: 0.967015 + +epoch 102 +train lattice loss: 0.571673 coord loss: 0.960625 + +epoch 103 +train lattice loss: 0.565716 coord loss: 0.967710 + +epoch 104 +train lattice loss: 0.574095 coord loss: 0.959632 + +epoch 105 +train lattice loss: 0.589011 coord loss: 0.972094 + +epoch 106 +train lattice loss: 0.577589 coord loss: 0.962948 + +epoch 107 +train lattice loss: 0.577674 coord loss: 0.965856 + +epoch 108 +train lattice loss: 0.570305 coord loss: 0.965661 + +epoch 109 +train lattice loss: 0.577285 coord loss: 0.958836 + +epoch 110 +train lattice loss: 0.575374 coord loss: 0.960430 +val lattice loss: 0.559048 coord loss: 0.974288 +save model with optimal loss + +epoch 111 +train lattice loss: 0.569693 coord loss: 0.967099 + +epoch 112 +train lattice loss: 0.576884 coord loss: 0.956861 + +epoch 113 +train lattice loss: 0.567020 coord loss: 0.968196 + +epoch 114 +train lattice loss: 0.568664 coord loss: 0.959039 + +epoch 115 +train lattice loss: 0.562282 coord loss: 0.960100 + +epoch 116 +train lattice loss: 0.569602 coord loss: 0.957680 + +epoch 117 +train lattice loss: 0.572689 coord loss: 0.956644 + +epoch 118 +train lattice loss: 0.561598 coord loss: 0.961071 + +epoch 119 +train lattice loss: 0.554585 coord loss: 0.966196 + +epoch 120 +train lattice loss: 0.564568 coord loss: 0.957477 +val lattice loss: 0.562121 coord loss: 0.963447 +save model with optimal loss + +epoch 121 +train lattice loss: 0.562311 coord loss: 0.959519 + +epoch 122 +train lattice loss: 0.559545 coord loss: 0.961399 + +epoch 123 +train lattice loss: 0.551952 coord loss: 0.959162 + +epoch 124 +train lattice loss: 0.569839 coord loss: 0.955839 + +epoch 125 +train lattice loss: 0.563055 coord loss: 0.960418 + +epoch 126 +train lattice loss: 0.567653 coord loss: 0.959133 + +epoch 127 +train lattice loss: 0.557269 coord loss: 0.957958 + +epoch 128 +train lattice loss: 0.561191 coord loss: 0.957698 + +epoch 129 +train lattice loss: 0.551808 coord loss: 0.958329 + +epoch 130 +train lattice loss: 0.556438 coord loss: 0.954301 +val lattice loss: 0.561389 coord loss: 0.960678 +save model with optimal loss + +epoch 131 +train lattice loss: 0.567160 coord loss: 0.958801 + +epoch 132 +train lattice loss: 0.571469 coord loss: 0.962390 + +epoch 133 +train lattice loss: 0.563499 coord loss: 0.956819 + +epoch 134 +train lattice loss: 0.564203 coord loss: 0.962440 + +epoch 135 +train lattice loss: 0.556121 coord loss: 0.958157 + +epoch 136 +train lattice loss: 0.554508 coord loss: 0.957604 + +epoch 137 +train lattice loss: 0.559024 coord loss: 0.962333 + +epoch 138 +train lattice loss: 0.559304 coord loss: 0.959020 + +epoch 139 +train lattice loss: 0.560461 coord loss: 0.960058 + +epoch 140 +train lattice loss: 0.561675 coord loss: 0.961086 +val lattice loss: 0.563033 coord loss: 0.960348 + +epoch 141 +train lattice loss: 0.562363 coord loss: 0.962143 + +epoch 142 +train lattice loss: 0.546772 coord loss: 0.955111 + +epoch 143 +train lattice loss: 0.594654 coord loss: 0.957305 + +epoch 144 +train lattice loss: 0.564745 coord loss: 0.961529 + +epoch 145 +train lattice loss: 0.553803 coord loss: 0.963249 + +epoch 146 +train lattice loss: 0.548685 coord loss: 0.957961 + +epoch 147 +train lattice loss: 0.550131 coord loss: 0.960970 + +epoch 148 +train lattice loss: 0.548891 coord loss: 0.958823 + +epoch 149 +train lattice loss: 0.550869 coord loss: 0.956351 + +epoch 150 +train lattice loss: 0.544238 coord loss: 0.957172 +val lattice loss: 0.556185 coord loss: 0.952604 +save model with optimal loss + +epoch 151 +train lattice loss: 0.556363 coord loss: 0.953152 + +epoch 152 +train lattice loss: 0.551299 coord loss: 0.961062 + +epoch 153 +train lattice loss: 0.549898 coord loss: 0.959901 + +epoch 154 +train lattice loss: 0.542926 coord loss: 0.958721 + +epoch 155 +train lattice loss: 0.547970 coord loss: 0.955324 + +epoch 156 +train lattice loss: 0.535035 coord loss: 0.955058 + +epoch 157 +train lattice loss: 0.548034 coord loss: 0.965338 + +epoch 158 +train lattice loss: 0.556787 coord loss: 0.948997 + +epoch 159 +train lattice loss: 0.541291 coord loss: 0.959267 + +epoch 160 +train lattice loss: 0.548730 coord loss: 0.946626 +val lattice loss: 0.526063 coord loss: 0.962858 +save model with optimal loss + +epoch 161 +train lattice loss: 0.548569 coord loss: 0.953536 + +epoch 162 +train lattice loss: 0.541338 coord loss: 0.950756 + +epoch 163 +train lattice loss: 0.551265 coord loss: 0.954368 + +epoch 164 +train lattice loss: 0.560056 coord loss: 0.953856 + +epoch 165 +train lattice loss: 0.565675 coord loss: 0.956932 + +epoch 166 +train lattice loss: 0.549682 coord loss: 0.957821 + +epoch 167 +train lattice loss: 0.549240 coord loss: 0.954790 + +epoch 168 +train lattice loss: 0.534226 coord loss: 0.954203 + +epoch 169 +train lattice loss: 0.536818 coord loss: 0.952777 + +epoch 170 +train lattice loss: 0.534505 coord loss: 0.954448 +val lattice loss: 0.558317 coord loss: 0.946403 + +epoch 171 +train lattice loss: 0.532361 coord loss: 0.955181 + +epoch 172 +train lattice loss: 0.536650 coord loss: 0.949588 + +epoch 173 +train lattice loss: 0.537536 coord loss: 0.950124 + +epoch 174 +train lattice loss: 0.522629 coord loss: 0.954329 + +epoch 175 +train lattice loss: 0.530629 coord loss: 0.949309 + +epoch 176 +train lattice loss: 0.535573 coord loss: 0.952238 + +epoch 177 +train lattice loss: 0.531516 coord loss: 0.951014 + +epoch 178 +train lattice loss: 0.534970 coord loss: 0.947365 + +epoch 179 +train lattice loss: 0.544088 coord loss: 0.952024 + +epoch 180 +train lattice loss: 0.530342 coord loss: 0.947033 +val lattice loss: 0.535664 coord loss: 0.954525 + +epoch 181 +train lattice loss: 0.529259 coord loss: 0.950091 + +epoch 182 +train lattice loss: 0.528842 coord loss: 0.957709 + +epoch 183 +train lattice loss: 0.529056 coord loss: 0.951881 + +epoch 184 +train lattice loss: 0.543542 coord loss: 0.947961 + +epoch 185 +train lattice loss: 0.536189 coord loss: 0.949722 + +epoch 186 +train lattice loss: 0.542052 coord loss: 0.947898 + +epoch 187 +train lattice loss: 0.526387 coord loss: 0.953072 + +epoch 188 +train lattice loss: 0.526870 coord loss: 0.952387 + +epoch 189 +train lattice loss: 0.542331 coord loss: 0.953766 + +epoch 190 +train lattice loss: 0.538424 coord loss: 0.950614 +val lattice loss: 0.555326 coord loss: 0.948202 + +epoch 191 +train lattice loss: 0.545573 coord loss: 0.956442 + +epoch 192 +train lattice loss: 0.542364 coord loss: 0.952233 + +epoch 193 +train lattice loss: 0.526866 coord loss: 0.955223 + +epoch 194 +train lattice loss: 0.536193 coord loss: 0.951433 + +epoch 195 +train lattice loss: 0.524919 coord loss: 0.949470 + +epoch 196 +train lattice loss: 0.540314 coord loss: 0.948608 + +epoch 197 +train lattice loss: 0.536547 coord loss: 0.954834 + +epoch 198 +train lattice loss: 0.532104 coord loss: 0.945763 + +epoch 199 +train lattice loss: 0.526467 coord loss: 0.946250 + +epoch 200 +train lattice loss: 0.536304 coord loss: 0.949683 +val lattice loss: 0.543603 coord loss: 0.952805 + +epoch 201 +train lattice loss: 0.531753 coord loss: 0.949193 + +epoch 202 +train lattice loss: 0.521300 coord loss: 0.950614 + +epoch 203 +train lattice loss: 0.536226 coord loss: 0.950320 + +epoch 204 +train lattice loss: 0.532561 coord loss: 0.946907 + +epoch 205 +train lattice loss: 0.536766 coord loss: 0.948619 + +epoch 206 +train lattice loss: 0.533479 coord loss: 0.945091 + +epoch 207 +train lattice loss: 0.541580 coord loss: 0.943011 + +epoch 208 +train lattice loss: 0.537582 coord loss: 0.944249 + +epoch 209 +train lattice loss: 0.522256 coord loss: 0.941876 + +epoch 210 +train lattice loss: 0.535625 coord loss: 0.945659 +val lattice loss: 0.533043 coord loss: 0.939599 +save model with optimal loss + +epoch 211 +train lattice loss: 0.532478 coord loss: 0.946635 + +epoch 212 +train lattice loss: 0.522547 coord loss: 0.943387 + +epoch 213 +train lattice loss: 0.523370 coord loss: 0.947659 + +epoch 214 +train lattice loss: 0.517074 coord loss: 0.940217 + +epoch 215 +train lattice loss: 0.518628 coord loss: 0.941331 + +epoch 216 +train lattice loss: 0.521036 coord loss: 0.948789 + +epoch 217 +train lattice loss: 0.518659 coord loss: 0.937475 + +epoch 218 +train lattice loss: 0.524771 coord loss: 0.951276 + +epoch 219 +train lattice loss: 0.524935 coord loss: 0.940128 + +epoch 220 +train lattice loss: 0.519628 coord loss: 0.945414 +val lattice loss: 0.519240 coord loss: 0.933105 +save model with optimal loss + +epoch 221 +train lattice loss: 0.525337 coord loss: 0.942768 + +epoch 222 +train lattice loss: 0.522586 coord loss: 0.943026 + +epoch 223 +train lattice loss: 0.516094 coord loss: 0.943351 + +epoch 224 +train lattice loss: 0.529750 coord loss: 0.940998 + +epoch 225 +train lattice loss: 0.525973 coord loss: 0.943794 + +epoch 226 +train lattice loss: 0.524063 coord loss: 0.942723 + +epoch 227 +train lattice loss: 0.524736 coord loss: 0.940719 + +epoch 228 +train lattice loss: 0.519075 coord loss: 0.945578 + +epoch 229 +train lattice loss: 0.533899 coord loss: 0.937935 + +epoch 230 +train lattice loss: 0.522929 coord loss: 0.946288 +val lattice loss: 0.513194 coord loss: 0.942145 + +epoch 231 +train lattice loss: 0.519304 coord loss: 0.943962 + +epoch 232 +train lattice loss: 0.525241 coord loss: 0.953908 + +epoch 233 +train lattice loss: 0.517748 coord loss: 0.939744 + +epoch 234 +train lattice loss: 0.532671 coord loss: 0.940526 + +epoch 235 +train lattice loss: 0.529476 coord loss: 0.940254 + +epoch 236 +train lattice loss: 0.518217 coord loss: 0.935759 + +epoch 237 +train lattice loss: 0.529072 coord loss: 0.940025 + +epoch 238 +train lattice loss: 0.534369 coord loss: 0.943048 + +epoch 239 +train lattice loss: 0.515370 coord loss: 0.939955 + +epoch 240 +train lattice loss: 0.509163 coord loss: 0.938455 +val lattice loss: 0.516059 coord loss: 0.935756 +save model with optimal loss + +epoch 241 +train lattice loss: 0.513927 coord loss: 0.945554 + +epoch 242 +train lattice loss: 0.520048 coord loss: 0.934081 + +epoch 243 +train lattice loss: 0.511698 coord loss: 0.942107 + +epoch 244 +train lattice loss: 0.524411 coord loss: 0.935503 + +epoch 245 +train lattice loss: 0.539133 coord loss: 0.933347 + +epoch 246 +train lattice loss: 0.529465 coord loss: 0.934674 + +epoch 247 +train lattice loss: 0.520663 coord loss: 0.936357 + +epoch 248 +train lattice loss: 0.531380 coord loss: 0.935164 + +epoch 249 +train lattice loss: 0.515428 coord loss: 0.932216 + +epoch 250 +train lattice loss: 0.510045 coord loss: 0.931402 +val lattice loss: 0.531343 coord loss: 0.931219 + +epoch 251 +train lattice loss: 0.514483 coord loss: 0.928293 + +epoch 252 +train lattice loss: 0.522020 coord loss: 0.938967 + +epoch 253 +train lattice loss: 0.515427 coord loss: 0.932951 + +epoch 254 +train lattice loss: 0.516971 coord loss: 0.932273 + +epoch 255 +train lattice loss: 0.518699 coord loss: 0.936827 + +epoch 256 +train lattice loss: 0.510347 coord loss: 0.936077 + +epoch 257 +train lattice loss: 0.521040 coord loss: 0.939216 + +epoch 258 +train lattice loss: 0.522887 coord loss: 0.931850 + +epoch 259 +train lattice loss: 0.518911 coord loss: 0.929930 + +epoch 260 +train lattice loss: 0.522190 coord loss: 0.932371 +val lattice loss: 0.511180 coord loss: 0.945830 + +epoch 261 +train lattice loss: 0.515841 coord loss: 0.935888 + +epoch 262 +train lattice loss: 0.499974 coord loss: 0.930243 + +epoch 263 +train lattice loss: 0.519223 coord loss: 0.930214 + +epoch 264 +train lattice loss: 0.524294 coord loss: 0.930431 + +epoch 265 +train lattice loss: 0.529077 coord loss: 0.933602 + +epoch 266 +train lattice loss: 0.516463 coord loss: 0.924823 + +epoch 267 +train lattice loss: 0.521658 coord loss: 0.933909 + +epoch 268 +train lattice loss: 0.518991 coord loss: 0.927747 + +epoch 269 +train lattice loss: 0.517468 coord loss: 0.928324 + +epoch 270 +train lattice loss: 0.512478 coord loss: 0.932036 +val lattice loss: 0.513310 coord loss: 0.926855 +save model with optimal loss + +epoch 271 +train lattice loss: 0.509823 coord loss: 0.926490 + +epoch 272 +train lattice loss: 0.512409 coord loss: 0.932413 + +epoch 273 +train lattice loss: 0.501966 coord loss: 0.927445 + +epoch 274 +train lattice loss: 0.506184 coord loss: 0.929894 + +epoch 275 +train lattice loss: 0.522528 coord loss: 0.922766 + +epoch 276 +train lattice loss: 0.515699 coord loss: 0.924768 + +epoch 277 +train lattice loss: 0.519801 coord loss: 0.924843 + +epoch 278 +train lattice loss: 0.514711 coord loss: 0.922758 + +epoch 279 +train lattice loss: 0.506909 coord loss: 0.926542 + +epoch 280 +train lattice loss: 0.513215 coord loss: 0.928255 +val lattice loss: 0.518887 coord loss: 0.926434 + +epoch 281 +train lattice loss: 0.520511 coord loss: 0.928367 + +epoch 282 +train lattice loss: 0.521113 coord loss: 0.921066 + +epoch 283 +train lattice loss: 0.515630 coord loss: 0.918867 + +epoch 284 +train lattice loss: 0.508594 coord loss: 0.912492 + +epoch 285 +train lattice loss: 0.516301 coord loss: 0.918518 + +epoch 286 +train lattice loss: 0.507929 coord loss: 0.925239 + +epoch 287 +train lattice loss: 0.509846 coord loss: 0.919067 + +epoch 288 +train lattice loss: 0.511326 coord loss: 0.924502 + +epoch 289 +train lattice loss: 0.526486 coord loss: 0.914967 + +epoch 290 +train lattice loss: 0.515511 coord loss: 0.926135 +val lattice loss: 0.535310 coord loss: 0.917559 + +epoch 291 +train lattice loss: 0.516830 coord loss: 0.919988 + +epoch 292 +train lattice loss: 0.516819 coord loss: 0.924181 + +epoch 293 +train lattice loss: 0.514911 coord loss: 0.915033 + +epoch 294 +train lattice loss: 0.508562 coord loss: 0.917657 + +epoch 295 +train lattice loss: 0.503793 coord loss: 0.916438 + +epoch 296 +train lattice loss: 0.508017 coord loss: 0.920736 + +epoch 297 +train lattice loss: 0.511142 coord loss: 0.916064 + +epoch 298 +train lattice loss: 0.512272 coord loss: 0.912697 + +epoch 299 +train lattice loss: 0.525187 coord loss: 0.918667 + +epoch 300 +train lattice loss: 0.515519 coord loss: 0.918492 +val lattice loss: 0.518039 coord loss: 0.911229 +save model with optimal loss + +epoch 301 +train lattice loss: 0.517025 coord loss: 0.912014 + +epoch 302 +train lattice loss: 0.517557 coord loss: 0.914998 + +epoch 303 +train lattice loss: 0.511907 coord loss: 0.914615 + +epoch 304 +train lattice loss: 0.508866 coord loss: 0.912261 + +epoch 305 +train lattice loss: 0.515602 coord loss: 0.912521 + +epoch 306 +train lattice loss: 0.517273 coord loss: 0.911449 + +epoch 307 +train lattice loss: 0.503994 coord loss: 0.914686 + +epoch 308 +train lattice loss: 0.508057 coord loss: 0.910461 + +epoch 309 +train lattice loss: 0.498512 coord loss: 0.916080 + +epoch 310 +train lattice loss: 0.505278 coord loss: 0.914589 +val lattice loss: 0.510750 coord loss: 0.913285 +save model with optimal loss + +epoch 311 +train lattice loss: 0.509826 coord loss: 0.913802 + +epoch 312 +train lattice loss: 0.517618 coord loss: 0.912843 + +epoch 313 +train lattice loss: 0.513453 coord loss: 0.906808 + +epoch 314 +train lattice loss: 0.509657 coord loss: 0.898112 + +epoch 315 +train lattice loss: 0.509623 coord loss: 0.909239 + +epoch 316 +train lattice loss: 0.516666 coord loss: 0.904650 + +epoch 317 +train lattice loss: 0.518964 coord loss: 0.901756 + +epoch 318 +train lattice loss: 0.524101 coord loss: 0.907313 + +epoch 319 +train lattice loss: 0.508999 coord loss: 0.907509 + +epoch 320 +train lattice loss: 0.496940 coord loss: 0.901444 +val lattice loss: 0.487340 coord loss: 0.903741 +save model with optimal loss + +epoch 321 +train lattice loss: 0.509009 coord loss: 0.900207 + +epoch 322 +train lattice loss: 0.508561 coord loss: 0.905868 + +epoch 323 +train lattice loss: 0.522438 coord loss: 0.900788 + +epoch 324 +train lattice loss: 0.510107 coord loss: 0.906226 + +epoch 325 +train lattice loss: 0.516758 coord loss: 0.900467 + +epoch 326 +train lattice loss: 0.503467 coord loss: 0.904903 + +epoch 327 +train lattice loss: 0.500372 coord loss: 0.910108 + +epoch 328 +train lattice loss: 0.506335 coord loss: 0.901938 + +epoch 329 +train lattice loss: 0.509409 coord loss: 0.904354 + +epoch 330 +train lattice loss: 0.506970 coord loss: 0.899307 +val lattice loss: 0.490003 coord loss: 0.892535 +save model with optimal loss + +epoch 331 +train lattice loss: 0.503056 coord loss: 0.893662 + +epoch 332 +train lattice loss: 0.499714 coord loss: 0.893013 + +epoch 333 +train lattice loss: 0.502550 coord loss: 0.899074 + +epoch 334 +train lattice loss: 0.500888 coord loss: 0.898318 + +epoch 335 +train lattice loss: 0.511171 coord loss: 0.895603 + +epoch 336 +train lattice loss: 0.497958 coord loss: 0.898154 + +epoch 337 +train lattice loss: 0.515214 coord loss: 0.895980 + +epoch 338 +train lattice loss: 0.514483 coord loss: 0.890731 + +epoch 339 +train lattice loss: 0.498904 coord loss: 0.892504 + +epoch 340 +train lattice loss: 0.509644 coord loss: 0.891956 +val lattice loss: 0.535407 coord loss: 0.880448 + +epoch 341 +train lattice loss: 0.511714 coord loss: 0.898214 + +epoch 342 +train lattice loss: 0.508800 coord loss: 0.897922 + +epoch 343 +train lattice loss: 0.495136 coord loss: 0.890280 + +epoch 344 +train lattice loss: 0.499731 coord loss: 0.894882 + +epoch 345 +train lattice loss: 0.507920 coord loss: 0.885692 + +epoch 346 +train lattice loss: 0.498978 coord loss: 0.893370 + +epoch 347 +train lattice loss: 0.505269 coord loss: 0.889388 + +epoch 348 +train lattice loss: 0.505630 coord loss: 0.891200 + +epoch 349 +train lattice loss: 0.501063 coord loss: 0.886412 + +epoch 350 +train lattice loss: 0.510923 coord loss: 0.887849 +val lattice loss: 0.486131 coord loss: 0.884517 +save model with optimal loss + +epoch 351 +train lattice loss: 0.506546 coord loss: 0.887222 + +epoch 352 +train lattice loss: 0.507284 coord loss: 0.885426 + +epoch 353 +train lattice loss: 0.502642 coord loss: 0.885494 + +epoch 354 +train lattice loss: 0.499321 coord loss: 0.887512 + +epoch 355 +train lattice loss: 0.496608 coord loss: 0.887318 + +epoch 356 +train lattice loss: 0.506285 coord loss: 0.883317 + +epoch 357 +train lattice loss: 0.498069 coord loss: 0.890171 + +epoch 358 +train lattice loss: 0.510949 coord loss: 0.888886 + +epoch 359 +train lattice loss: 0.506559 coord loss: 0.880822 + +epoch 360 +train lattice loss: 0.498216 coord loss: 0.883927 +val lattice loss: 0.490516 coord loss: 0.892075 + +epoch 361 +train lattice loss: 0.496863 coord loss: 0.884289 + +epoch 362 +train lattice loss: 0.505739 coord loss: 0.885783 + +epoch 363 +train lattice loss: 0.512083 coord loss: 0.879845 + +epoch 364 +train lattice loss: 0.516241 coord loss: 0.884053 + +epoch 365 +train lattice loss: 0.504726 coord loss: 0.884184 + +epoch 366 +train lattice loss: 0.501928 coord loss: 0.878355 + +epoch 367 +train lattice loss: 0.496755 coord loss: 0.886087 + +epoch 368 +train lattice loss: 0.501486 coord loss: 0.883141 + +epoch 369 +train lattice loss: 0.510263 coord loss: 0.878466 + +epoch 370 +train lattice loss: 0.502201 coord loss: 0.881368 +val lattice loss: 0.507217 coord loss: 0.889032 + +epoch 371 +train lattice loss: 0.500837 coord loss: 0.877563 + +epoch 372 +train lattice loss: 0.493764 coord loss: 0.874008 + +epoch 373 +train lattice loss: 0.487556 coord loss: 0.875842 + +epoch 374 +train lattice loss: 0.505128 coord loss: 0.876282 + +epoch 375 +train lattice loss: 0.497587 coord loss: 0.872284 + +epoch 376 +train lattice loss: 0.501386 coord loss: 0.880571 + +epoch 377 +train lattice loss: 0.505774 coord loss: 0.877451 + +epoch 378 +train lattice loss: 0.490042 coord loss: 0.883742 + +epoch 379 +train lattice loss: 0.492282 coord loss: 0.876985 + +epoch 380 +train lattice loss: 0.503290 coord loss: 0.878545 +val lattice loss: 0.514972 coord loss: 0.875238 + +epoch 381 +train lattice loss: 0.506256 coord loss: 0.871430 + +epoch 382 +train lattice loss: 0.499362 coord loss: 0.872952 + +epoch 383 +train lattice loss: 0.503360 coord loss: 0.873407 + +epoch 384 +train lattice loss: 0.500374 coord loss: 0.869776 + +epoch 385 +train lattice loss: 0.506897 coord loss: 0.876877 + +epoch 386 +train lattice loss: 0.509738 coord loss: 0.875312 + +epoch 387 +train lattice loss: 0.500033 coord loss: 0.875246 + +epoch 388 +train lattice loss: 0.497995 coord loss: 0.876714 + +epoch 389 +train lattice loss: 0.501702 coord loss: 0.873262 + +epoch 390 +train lattice loss: 0.506950 coord loss: 0.870405 +val lattice loss: 0.495538 coord loss: 0.870041 +save model with optimal loss + +epoch 391 +train lattice loss: 0.492378 coord loss: 0.867024 + +epoch 392 +train lattice loss: 0.491964 coord loss: 0.869180 + +epoch 393 +train lattice loss: 0.496956 coord loss: 0.864127 + +epoch 394 +train lattice loss: 0.492401 coord loss: 0.869138 + +epoch 395 +train lattice loss: 0.495256 coord loss: 0.871328 + +epoch 396 +train lattice loss: 0.495891 coord loss: 0.871260 + +epoch 397 +train lattice loss: 0.501825 coord loss: 0.870971 + +epoch 398 +train lattice loss: 0.490207 coord loss: 0.868323 + +epoch 399 +train lattice loss: 0.497541 coord loss: 0.867342 + +epoch 400 +train lattice loss: 0.497279 coord loss: 0.867577 +val lattice loss: 0.481313 coord loss: 0.874976 +save model with optimal loss + +epoch 401 +train lattice loss: 0.494940 coord loss: 0.866147 + +epoch 402 +train lattice loss: 0.489138 coord loss: 0.867644 + +epoch 403 +train lattice loss: 0.486909 coord loss: 0.871404 + +epoch 404 +train lattice loss: 0.499095 coord loss: 0.860716 + +epoch 405 +train lattice loss: 0.493228 coord loss: 0.870119 + +epoch 406 +train lattice loss: 0.490156 coord loss: 0.861914 + +epoch 407 +train lattice loss: 0.494405 coord loss: 0.863526 + +epoch 408 +train lattice loss: 0.496041 coord loss: 0.866607 + +epoch 409 +train lattice loss: 0.485846 coord loss: 0.859696 + +epoch 410 +train lattice loss: 0.493177 coord loss: 0.865167 +val lattice loss: 0.483522 coord loss: 0.863085 +save model with optimal loss + +epoch 411 +train lattice loss: 0.488982 coord loss: 0.864536 + +epoch 412 +train lattice loss: 0.492391 coord loss: 0.866677 + +epoch 413 +train lattice loss: 0.495080 coord loss: 0.865314 + +epoch 414 +train lattice loss: 0.493208 coord loss: 0.869100 + +epoch 415 +train lattice loss: 0.499565 coord loss: 0.858971 + +epoch 416 +train lattice loss: 0.489367 coord loss: 0.862764 + +epoch 417 +train lattice loss: 0.482467 coord loss: 0.860613 + +epoch 418 +train lattice loss: 0.495151 coord loss: 0.858010 + +epoch 419 +train lattice loss: 0.497128 coord loss: 0.858746 + +epoch 420 +train lattice loss: 0.486057 coord loss: 0.864626 +val lattice loss: 0.504286 coord loss: 0.869425 + +epoch 421 +train lattice loss: 0.490093 coord loss: 0.863654 + +epoch 422 +train lattice loss: 0.503942 coord loss: 0.862143 + +epoch 423 +train lattice loss: 0.488549 coord loss: 0.855792 + +epoch 424 +train lattice loss: 0.494348 coord loss: 0.851328 + +epoch 425 +train lattice loss: 0.486695 coord loss: 0.856345 + +epoch 426 +train lattice loss: 0.492431 coord loss: 0.854287 + +epoch 427 +train lattice loss: 0.496140 coord loss: 0.857842 + +epoch 428 +train lattice loss: 0.487606 coord loss: 0.858684 + +epoch 429 +train lattice loss: 0.492859 coord loss: 0.860737 + +epoch 430 +train lattice loss: 0.492014 coord loss: 0.861037 +val lattice loss: 0.497055 coord loss: 0.861261 + +epoch 431 +train lattice loss: 0.493377 coord loss: 0.855041 + +epoch 432 +train lattice loss: 0.485989 coord loss: 0.854772 + +epoch 433 +train lattice loss: 0.491751 coord loss: 0.853897 + +epoch 434 +train lattice loss: 0.501639 coord loss: 0.853436 + +epoch 435 +train lattice loss: 0.498855 coord loss: 0.856474 + +epoch 436 +train lattice loss: 0.483462 coord loss: 0.858108 + +epoch 437 +train lattice loss: 0.497204 coord loss: 0.854353 + +epoch 438 +train lattice loss: 0.497256 coord loss: 0.854527 + +epoch 439 +train lattice loss: 0.485878 coord loss: 0.854992 + +epoch 440 +train lattice loss: 0.496007 coord loss: 0.852339 +val lattice loss: 0.500061 coord loss: 0.859896 + +epoch 441 +train lattice loss: 0.493914 coord loss: 0.852706 + +epoch 442 +train lattice loss: 0.498154 coord loss: 0.853079 + +epoch 443 +train lattice loss: 0.494397 coord loss: 0.857699 + +epoch 444 +train lattice loss: 0.491751 coord loss: 0.850672 + +epoch 445 +train lattice loss: 0.483499 coord loss: 0.853624 + +epoch 446 +train lattice loss: 0.482689 coord loss: 0.856570 + +epoch 447 +train lattice loss: 0.494330 coord loss: 0.854141 + +epoch 448 +train lattice loss: 0.489625 coord loss: 0.857086 + +epoch 449 +train lattice loss: 0.489145 coord loss: 0.848664 + +epoch 450 +train lattice loss: 0.487440 coord loss: 0.849730 +val lattice loss: 0.492159 coord loss: 0.854801 + +epoch 451 +train lattice loss: 0.474433 coord loss: 0.850594 + +epoch 452 +train lattice loss: 0.482268 coord loss: 0.849935 + +epoch 453 +train lattice loss: 0.485768 coord loss: 0.845789 + +epoch 454 +train lattice loss: 0.490876 coord loss: 0.844594 + +epoch 455 +train lattice loss: 0.495417 coord loss: 0.851054 + +epoch 456 +train lattice loss: 0.489599 coord loss: 0.845984 + +epoch 457 +train lattice loss: 0.489323 coord loss: 0.843732 + +epoch 458 +train lattice loss: 0.484564 coord loss: 0.851159 + +epoch 459 +train lattice loss: 0.482994 coord loss: 0.851664 + +epoch 460 +train lattice loss: 0.493031 coord loss: 0.847597 +val lattice loss: 0.472952 coord loss: 0.848656 +save model with optimal loss + +epoch 461 +train lattice loss: 0.490944 coord loss: 0.845754 + +epoch 462 +train lattice loss: 0.481255 coord loss: 0.849655 + +epoch 463 +train lattice loss: 0.502853 coord loss: 0.845548 + +epoch 464 +train lattice loss: 0.480989 coord loss: 0.849156 + +epoch 465 +train lattice loss: 0.482546 coord loss: 0.845808 + +epoch 466 +train lattice loss: 0.486183 coord loss: 0.840963 + +epoch 467 +train lattice loss: 0.485639 coord loss: 0.847963 + +epoch 468 +train lattice loss: 0.482043 coord loss: 0.845440 + +epoch 469 +train lattice loss: 0.486790 coord loss: 0.844311 + +epoch 470 +train lattice loss: 0.475230 coord loss: 0.845853 +val lattice loss: 0.485217 coord loss: 0.834563 +save model with optimal loss + +epoch 471 +train lattice loss: 0.486101 coord loss: 0.844478 + +epoch 472 +train lattice loss: 0.488473 coord loss: 0.849954 + +epoch 473 +train lattice loss: 0.481058 coord loss: 0.844890 + +epoch 474 +train lattice loss: 0.476629 coord loss: 0.852677 + +epoch 475 +train lattice loss: 0.488002 coord loss: 0.839988 + +epoch 476 +train lattice loss: 0.478761 coord loss: 0.843593 + +epoch 477 +train lattice loss: 0.487260 coord loss: 0.844690 + +epoch 478 +train lattice loss: 0.480175 coord loss: 0.846289 + +epoch 479 +train lattice loss: 0.476922 coord loss: 0.838933 + +epoch 480 +train lattice loss: 0.489998 coord loss: 0.838651 +val lattice loss: 0.494280 coord loss: 0.856076 + +epoch 481 +train lattice loss: 0.482345 coord loss: 0.841539 + +epoch 482 +train lattice loss: 0.490670 coord loss: 0.841749 + +epoch 483 +train lattice loss: 0.483157 coord loss: 0.848476 + +epoch 484 +train lattice loss: 0.484945 coord loss: 0.842429 + +epoch 485 +train lattice loss: 0.493325 coord loss: 0.842731 + +epoch 486 +train lattice loss: 0.488740 coord loss: 0.840349 + +epoch 487 +train lattice loss: 0.477670 coord loss: 0.841218 + +epoch 488 +train lattice loss: 0.482966 coord loss: 0.836967 + +epoch 489 +train lattice loss: 0.492055 coord loss: 0.837036 + +epoch 490 +train lattice loss: 0.475273 coord loss: 0.834003 +val lattice loss: 0.493306 coord loss: 0.834353 + +epoch 491 +train lattice loss: 0.481558 coord loss: 0.836081 + +epoch 492 +train lattice loss: 0.484018 coord loss: 0.840391 + +epoch 493 +train lattice loss: 0.483408 coord loss: 0.838056 + +epoch 494 +train lattice loss: 0.491733 coord loss: 0.839032 + +epoch 495 +train lattice loss: 0.485366 coord loss: 0.836634 + +epoch 496 +train lattice loss: 0.482345 coord loss: 0.837694 + +epoch 497 +train lattice loss: 0.476466 coord loss: 0.839709 + +epoch 498 +train lattice loss: 0.484763 coord loss: 0.837693 + +epoch 499 +train lattice loss: 0.484242 coord loss: 0.838480 + +epoch 500 +train lattice loss: 0.478362 coord loss: 0.840022 +val lattice loss: 0.466853 coord loss: 0.834901 +save model with optimal loss + +epoch 501 +train lattice loss: 0.475676 coord loss: 0.835594 + +epoch 502 +train lattice loss: 0.483517 coord loss: 0.830388 + +epoch 503 +train lattice loss: 0.472136 coord loss: 0.844021 + +epoch 504 +train lattice loss: 0.482327 coord loss: 0.837981 + +epoch 505 +train lattice loss: 0.479774 coord loss: 0.839188 + +epoch 506 +train lattice loss: 0.480976 coord loss: 0.833777 + +epoch 507 +train lattice loss: 0.478017 coord loss: 0.834380 + +epoch 508 +train lattice loss: 0.478616 coord loss: 0.827460 + +epoch 509 +train lattice loss: 0.481503 coord loss: 0.831334 + +epoch 510 +train lattice loss: 0.477110 coord loss: 0.829084 +val lattice loss: 0.469201 coord loss: 0.831179 +save model with optimal loss + +epoch 511 +train lattice loss: 0.479922 coord loss: 0.828963 + +epoch 512 +train lattice loss: 0.465458 coord loss: 0.829566 + +epoch 513 +train lattice loss: 0.470598 coord loss: 0.829160 + +epoch 514 +train lattice loss: 0.475620 coord loss: 0.829002 + +epoch 515 +train lattice loss: 0.482974 coord loss: 0.829166 + +epoch 516 +train lattice loss: 0.481234 coord loss: 0.832083 + +epoch 517 +train lattice loss: 0.482684 coord loss: 0.832362 + +epoch 518 +train lattice loss: 0.478806 coord loss: 0.838890 + +epoch 519 +train lattice loss: 0.473184 coord loss: 0.832422 + +epoch 520 +train lattice loss: 0.482005 coord loss: 0.826575 +val lattice loss: 0.479778 coord loss: 0.826920 + +epoch 521 +train lattice loss: 0.468630 coord loss: 0.832413 + +epoch 522 +train lattice loss: 0.477157 coord loss: 0.829786 + +epoch 523 +train lattice loss: 0.481027 coord loss: 0.829382 + +epoch 524 +train lattice loss: 0.476778 coord loss: 0.828322 + +epoch 525 +train lattice loss: 0.476627 coord loss: 0.825079 + +epoch 526 +train lattice loss: 0.482270 coord loss: 0.826886 + +epoch 527 +train lattice loss: 0.479336 coord loss: 0.829740 + +epoch 528 +train lattice loss: 0.490547 coord loss: 0.824648 + +epoch 529 +train lattice loss: 0.473113 coord loss: 0.826701 + +epoch 530 +train lattice loss: 0.482929 coord loss: 0.831609 +val lattice loss: 0.488445 coord loss: 0.829401 + +epoch 531 +train lattice loss: 0.483164 coord loss: 0.823381 + +epoch 532 +train lattice loss: 0.482032 coord loss: 0.825033 + +epoch 533 +train lattice loss: 0.479411 coord loss: 0.823709 + +epoch 534 +train lattice loss: 0.473266 coord loss: 0.825157 + +epoch 535 +train lattice loss: 0.464361 coord loss: 0.824661 + +epoch 536 +train lattice loss: 0.475124 coord loss: 0.827183 + +epoch 537 +train lattice loss: 0.466546 coord loss: 0.821353 + +epoch 538 +train lattice loss: 0.468974 coord loss: 0.821556 + +epoch 539 +train lattice loss: 0.477730 coord loss: 0.824731 + +epoch 540 +train lattice loss: 0.481160 coord loss: 0.823980 +val lattice loss: 0.485803 coord loss: 0.814991 + +epoch 541 +train lattice loss: 0.473441 coord loss: 0.824771 + +epoch 542 +train lattice loss: 0.474151 coord loss: 0.822571 + +epoch 543 +train lattice loss: 0.474649 coord loss: 0.819788 + +epoch 544 +train lattice loss: 0.465734 coord loss: 0.826818 + +epoch 545 +train lattice loss: 0.479079 coord loss: 0.817990 + +epoch 546 +train lattice loss: 0.475706 coord loss: 0.818631 + +epoch 547 +train lattice loss: 0.464646 coord loss: 0.817985 + +epoch 548 +train lattice loss: 0.473134 coord loss: 0.822349 + +epoch 549 +train lattice loss: 0.464735 coord loss: 0.821834 + +epoch 550 +train lattice loss: 0.464999 coord loss: 0.819599 +val lattice loss: 0.464772 coord loss: 0.831455 +save model with optimal loss + +epoch 551 +train lattice loss: 0.475295 coord loss: 0.821895 + +epoch 552 +train lattice loss: 0.468063 coord loss: 0.815197 + +epoch 553 +train lattice loss: 0.478575 coord loss: 0.820347 + +epoch 554 +train lattice loss: 0.470804 coord loss: 0.820749 + +epoch 555 +train lattice loss: 0.473974 coord loss: 0.816372 + +epoch 556 +train lattice loss: 0.475870 coord loss: 0.815630 + +epoch 557 +train lattice loss: 0.475169 coord loss: 0.819314 + +epoch 558 +train lattice loss: 0.470694 coord loss: 0.821002 + +epoch 559 +train lattice loss: 0.456906 coord loss: 0.815784 + +epoch 560 +train lattice loss: 0.479727 coord loss: 0.820882 +val lattice loss: 0.469463 coord loss: 0.815266 +save model with optimal loss + +epoch 561 +train lattice loss: 0.466085 coord loss: 0.820529 + +epoch 562 +train lattice loss: 0.469708 coord loss: 0.818804 + +epoch 563 +train lattice loss: 0.472033 coord loss: 0.816351 + +epoch 564 +train lattice loss: 0.457769 coord loss: 0.817216 + +epoch 565 +train lattice loss: 0.464696 coord loss: 0.812647 + +epoch 566 +train lattice loss: 0.467170 coord loss: 0.817301 + +epoch 567 +train lattice loss: 0.463605 coord loss: 0.813253 + +epoch 568 +train lattice loss: 0.466456 coord loss: 0.812192 + +epoch 569 +train lattice loss: 0.463428 coord loss: 0.819889 + +epoch 570 +train lattice loss: 0.473152 coord loss: 0.814411 +val lattice loss: 0.482430 coord loss: 0.814596 + +epoch 571 +train lattice loss: 0.476769 coord loss: 0.816952 + +epoch 572 +train lattice loss: 0.473464 coord loss: 0.815108 + +epoch 573 +train lattice loss: 0.470856 coord loss: 0.816318 + +epoch 574 +train lattice loss: 0.461110 coord loss: 0.815320 + +epoch 575 +train lattice loss: 0.457812 coord loss: 0.816968 + +epoch 576 +train lattice loss: 0.472442 coord loss: 0.818296 + +epoch 577 +train lattice loss: 0.471704 coord loss: 0.810415 + +epoch 578 +train lattice loss: 0.466972 coord loss: 0.815279 + +epoch 579 +train lattice loss: 0.471504 coord loss: 0.816482 + +epoch 580 +train lattice loss: 0.475026 coord loss: 0.810507 +val lattice loss: 0.476762 coord loss: 0.811811 + +epoch 581 +train lattice loss: 0.467259 coord loss: 0.810360 + +epoch 582 +train lattice loss: 0.464337 coord loss: 0.812148 + +epoch 583 +train lattice loss: 0.459801 coord loss: 0.817880 + +epoch 584 +train lattice loss: 0.459790 coord loss: 0.808538 + +epoch 585 +train lattice loss: 0.466160 coord loss: 0.808047 + +epoch 586 +train lattice loss: 0.467321 coord loss: 0.811046 + +epoch 587 +train lattice loss: 0.461852 coord loss: 0.803038 + +epoch 588 +train lattice loss: 0.478353 coord loss: 0.805906 + +epoch 589 +train lattice loss: 0.464280 coord loss: 0.807636 + +epoch 590 +train lattice loss: 0.451189 coord loss: 0.806491 +val lattice loss: 0.455382 coord loss: 0.811233 +save model with optimal loss + +epoch 591 +train lattice loss: 0.464498 coord loss: 0.805653 + +epoch 592 +train lattice loss: 0.461469 coord loss: 0.804498 + +epoch 593 +train lattice loss: 0.470086 coord loss: 0.808974 + +epoch 594 +train lattice loss: 0.463893 coord loss: 0.801189 + +epoch 595 +train lattice loss: 0.477090 coord loss: 0.801665 + +epoch 596 +train lattice loss: 0.468679 coord loss: 0.802653 + +epoch 597 +train lattice loss: 0.476707 coord loss: 0.809423 + +epoch 598 +train lattice loss: 0.478552 coord loss: 0.807044 + +epoch 599 +train lattice loss: 0.470585 coord loss: 0.809626 + +epoch 600 +train lattice loss: 0.457881 coord loss: 0.810659 +val lattice loss: 0.466624 coord loss: 0.808658 + +epoch 601 +train lattice loss: 0.465613 coord loss: 0.805321 + +epoch 602 +train lattice loss: 0.465920 coord loss: 0.803200 + +epoch 603 +train lattice loss: 0.469178 coord loss: 0.808059 + +epoch 604 +train lattice loss: 0.464975 coord loss: 0.803929 + +epoch 605 +train lattice loss: 0.467227 coord loss: 0.805073 + +epoch 606 +train lattice loss: 0.461719 coord loss: 0.811407 + +epoch 607 +train lattice loss: 0.457108 coord loss: 0.799045 + +epoch 608 +train lattice loss: 0.458385 coord loss: 0.800629 + +epoch 609 +train lattice loss: 0.459627 coord loss: 0.796291 + +epoch 610 +train lattice loss: 0.458476 coord loss: 0.800173 +val lattice loss: 0.448171 coord loss: 0.802100 +save model with optimal loss + +epoch 611 +train lattice loss: 0.462617 coord loss: 0.794663 + +epoch 612 +train lattice loss: 0.455687 coord loss: 0.801422 + +epoch 613 +train lattice loss: 0.459497 coord loss: 0.796213 + +epoch 614 +train lattice loss: 0.460069 coord loss: 0.803011 + +epoch 615 +train lattice loss: 0.455329 coord loss: 0.800655 + +epoch 616 +train lattice loss: 0.468581 coord loss: 0.802815 + +epoch 617 +train lattice loss: 0.459484 coord loss: 0.805684 + +epoch 618 +train lattice loss: 0.458744 coord loss: 0.806534 + +epoch 619 +train lattice loss: 0.458316 coord loss: 0.796432 + +epoch 620 +train lattice loss: 0.463750 coord loss: 0.798859 +val lattice loss: 0.463486 coord loss: 0.799957 + +epoch 621 +train lattice loss: 0.470216 coord loss: 0.796127 + +epoch 622 +train lattice loss: 0.468990 coord loss: 0.801453 + +epoch 623 +train lattice loss: 0.462861 coord loss: 0.798803 + +epoch 624 +train lattice loss: 0.463463 coord loss: 0.801248 + +epoch 625 +train lattice loss: 0.460436 coord loss: 0.796493 + +epoch 626 +train lattice loss: 0.459194 coord loss: 0.797926 + +epoch 627 +train lattice loss: 0.463488 coord loss: 0.799208 + +epoch 628 +train lattice loss: 0.457833 coord loss: 0.796232 + +epoch 629 +train lattice loss: 0.455294 coord loss: 0.798669 + +epoch 630 +train lattice loss: 0.467985 coord loss: 0.790036 +val lattice loss: 0.457425 coord loss: 0.799789 + +epoch 631 +train lattice loss: 0.456951 coord loss: 0.799908 + +epoch 632 +train lattice loss: 0.454275 coord loss: 0.794096 + +epoch 633 +train lattice loss: 0.451519 coord loss: 0.788469 + +epoch 634 +train lattice loss: 0.456719 coord loss: 0.791911 + +epoch 635 +train lattice loss: 0.453162 coord loss: 0.802033 + +epoch 636 +train lattice loss: 0.446615 coord loss: 0.790268 + +epoch 637 +train lattice loss: 0.457546 coord loss: 0.793696 + +epoch 638 +train lattice loss: 0.457526 coord loss: 0.788181 + +epoch 639 +train lattice loss: 0.450695 coord loss: 0.792482 + +epoch 640 +train lattice loss: 0.458831 coord loss: 0.794741 +val lattice loss: 0.454248 coord loss: 0.798685 + +epoch 641 +train lattice loss: 0.464792 coord loss: 0.782261 + +epoch 642 +train lattice loss: 0.466608 coord loss: 0.791926 + +epoch 643 +train lattice loss: 0.461329 coord loss: 0.798441 + +epoch 644 +train lattice loss: 0.463789 coord loss: 0.792299 + +epoch 645 +train lattice loss: 0.457756 coord loss: 0.792415 + +epoch 646 +train lattice loss: 0.450381 coord loss: 0.785165 + +epoch 647 +train lattice loss: 0.450484 coord loss: 0.784630 + +epoch 648 +train lattice loss: 0.445456 coord loss: 0.789124 + +epoch 649 +train lattice loss: 0.465543 coord loss: 0.791887 + +epoch 650 +train lattice loss: 0.460396 coord loss: 0.793403 +val lattice loss: 0.452685 coord loss: 0.795724 +save model with optimal loss + +epoch 651 +train lattice loss: 0.452555 coord loss: 0.787057 + +epoch 652 +train lattice loss: 0.448745 coord loss: 0.790993 + +epoch 653 +train lattice loss: 0.455858 coord loss: 0.786174 + +epoch 654 +train lattice loss: 0.455191 coord loss: 0.787146 + +epoch 655 +train lattice loss: 0.456375 coord loss: 0.784998 + +epoch 656 +train lattice loss: 0.456683 coord loss: 0.786130 + +epoch 657 +train lattice loss: 0.466731 coord loss: 0.786585 + +epoch 658 +train lattice loss: 0.463085 coord loss: 0.783896 + +epoch 659 +train lattice loss: 0.460469 coord loss: 0.787066 + +epoch 660 +train lattice loss: 0.453410 coord loss: 0.788544 +val lattice loss: 0.460846 coord loss: 0.786219 +save model with optimal loss + +epoch 661 +train lattice loss: 0.455092 coord loss: 0.783908 + +epoch 662 +train lattice loss: 0.448195 coord loss: 0.779336 + +epoch 663 +train lattice loss: 0.454671 coord loss: 0.787740 + +epoch 664 +train lattice loss: 0.453539 coord loss: 0.776646 + +epoch 665 +train lattice loss: 0.447284 coord loss: 0.781024 + +epoch 666 +train lattice loss: 0.449121 coord loss: 0.780135 + +epoch 667 +train lattice loss: 0.467277 coord loss: 0.777891 + +epoch 668 +train lattice loss: 0.455008 coord loss: 0.781066 + +epoch 669 +train lattice loss: 0.452698 coord loss: 0.782856 + +epoch 670 +train lattice loss: 0.462409 coord loss: 0.783233 +val lattice loss: 0.457452 coord loss: 0.787153 +save model with optimal loss + +epoch 671 +train lattice loss: 0.450113 coord loss: 0.781756 + +epoch 672 +train lattice loss: 0.448777 coord loss: 0.774513 + +epoch 673 +train lattice loss: 0.459611 coord loss: 0.779111 + +epoch 674 +train lattice loss: 0.462619 coord loss: 0.782559 + +epoch 675 +train lattice loss: 0.465733 coord loss: 0.777569 + +epoch 676 +train lattice loss: 0.460698 coord loss: 0.778937 + +epoch 677 +train lattice loss: 0.451020 coord loss: 0.782178 + +epoch 678 +train lattice loss: 0.453408 coord loss: 0.775287 + +epoch 679 +train lattice loss: 0.456798 coord loss: 0.780033 + +epoch 680 +train lattice loss: 0.450266 coord loss: 0.776377 +val lattice loss: 0.440760 coord loss: 0.788424 +save model with optimal loss + +epoch 681 +train lattice loss: 0.457763 coord loss: 0.777651 + +epoch 682 +train lattice loss: 0.455375 coord loss: 0.782570 + +epoch 683 +train lattice loss: 0.454004 coord loss: 0.768976 + +epoch 684 +train lattice loss: 0.452369 coord loss: 0.778182 + +epoch 685 +train lattice loss: 0.457104 coord loss: 0.770110 + +epoch 686 +train lattice loss: 0.454452 coord loss: 0.766164 + +epoch 687 +train lattice loss: 0.457999 coord loss: 0.776550 + +epoch 688 +train lattice loss: 0.450989 coord loss: 0.770127 + +epoch 689 +train lattice loss: 0.454452 coord loss: 0.770169 + +epoch 690 +train lattice loss: 0.448031 coord loss: 0.776669 +val lattice loss: 0.463807 coord loss: 0.776404 + +epoch 691 +train lattice loss: 0.451264 coord loss: 0.770792 + +epoch 692 +train lattice loss: 0.443976 coord loss: 0.781055 + +epoch 693 +train lattice loss: 0.460396 coord loss: 0.778600 + +epoch 694 +train lattice loss: 0.447954 coord loss: 0.771821 + +epoch 695 +train lattice loss: 0.446741 coord loss: 0.766984 + +epoch 696 +train lattice loss: 0.452550 coord loss: 0.769651 + +epoch 697 +train lattice loss: 0.442275 coord loss: 0.767815 + +epoch 698 +train lattice loss: 0.450723 coord loss: 0.767637 + +epoch 699 +train lattice loss: 0.457581 coord loss: 0.771473 + +epoch 700 +train lattice loss: 0.448915 coord loss: 0.768201 +val lattice loss: 0.445029 coord loss: 0.768941 +save model with optimal loss + +epoch 701 +train lattice loss: 0.453860 coord loss: 0.770004 + +epoch 702 +train lattice loss: 0.448030 coord loss: 0.772681 + +epoch 703 +train lattice loss: 0.453696 coord loss: 0.768721 + +epoch 704 +train lattice loss: 0.457354 coord loss: 0.762683 + +epoch 705 +train lattice loss: 0.449210 coord loss: 0.768548 + +epoch 706 +train lattice loss: 0.453568 coord loss: 0.762816 + +epoch 707 +train lattice loss: 0.451533 coord loss: 0.766315 + +epoch 708 +train lattice loss: 0.450168 coord loss: 0.775132 + +epoch 709 +train lattice loss: 0.448220 coord loss: 0.765964 + +epoch 710 +train lattice loss: 0.456729 coord loss: 0.761810 +val lattice loss: 0.461962 coord loss: 0.764001 + +epoch 711 +train lattice loss: 0.443291 coord loss: 0.771325 + +epoch 712 +train lattice loss: 0.447813 coord loss: 0.768493 + +epoch 713 +train lattice loss: 0.448658 coord loss: 0.764712 + +epoch 714 +train lattice loss: 0.449580 coord loss: 0.763249 + +epoch 715 +train lattice loss: 0.451081 coord loss: 0.767639 + +epoch 716 +train lattice loss: 0.442381 coord loss: 0.766182 + +epoch 717 +train lattice loss: 0.442665 coord loss: 0.768691 + +epoch 718 +train lattice loss: 0.446700 coord loss: 0.760396 + +epoch 719 +train lattice loss: 0.454085 coord loss: 0.762285 + +epoch 720 +train lattice loss: 0.446351 coord loss: 0.763297 +val lattice loss: 0.452085 coord loss: 0.762948 + +epoch 721 +train lattice loss: 0.443281 coord loss: 0.762449 + +epoch 722 +train lattice loss: 0.450416 coord loss: 0.756628 + +epoch 723 +train lattice loss: 0.447960 coord loss: 0.749320 + +epoch 724 +train lattice loss: 0.456141 coord loss: 0.758072 + +epoch 725 +train lattice loss: 0.445071 coord loss: 0.759243 + +epoch 726 +train lattice loss: 0.450642 coord loss: 0.760046 + +epoch 727 +train lattice loss: 0.449163 coord loss: 0.762109 + +epoch 728 +train lattice loss: 0.436577 coord loss: 0.763797 + +epoch 729 +train lattice loss: 0.447673 coord loss: 0.750916 + +epoch 730 +train lattice loss: 0.449345 coord loss: 0.758686 +val lattice loss: 0.434792 coord loss: 0.761361 +save model with optimal loss + +epoch 731 +train lattice loss: 0.438610 coord loss: 0.754508 + +epoch 732 +train lattice loss: 0.449618 coord loss: 0.755988 + +epoch 733 +train lattice loss: 0.447335 coord loss: 0.761651 + +epoch 734 +train lattice loss: 0.453285 coord loss: 0.759593 + +epoch 735 +train lattice loss: 0.445557 coord loss: 0.755290 + +epoch 736 +train lattice loss: 0.447315 coord loss: 0.759590 + +epoch 737 +train lattice loss: 0.453992 coord loss: 0.745584 + +epoch 738 +train lattice loss: 0.442292 coord loss: 0.759937 + +epoch 739 +train lattice loss: 0.454405 coord loss: 0.755237 + +epoch 740 +train lattice loss: 0.450050 coord loss: 0.749343 +val lattice loss: 0.433196 coord loss: 0.767777 + +epoch 741 +train lattice loss: 0.448754 coord loss: 0.747933 + +epoch 742 +train lattice loss: 0.442861 coord loss: 0.752186 + +epoch 743 +train lattice loss: 0.447351 coord loss: 0.753276 + +epoch 744 +train lattice loss: 0.446343 coord loss: 0.756568 + +epoch 745 +train lattice loss: 0.449011 coord loss: 0.749989 + +epoch 746 +train lattice loss: 0.450201 coord loss: 0.753953 + +epoch 747 +train lattice loss: 0.447639 coord loss: 0.750115 + +epoch 748 +train lattice loss: 0.438938 coord loss: 0.749253 + +epoch 749 +train lattice loss: 0.447237 coord loss: 0.749028 + +epoch 750 +train lattice loss: 0.461641 coord loss: 0.748406 +val lattice loss: 0.442812 coord loss: 0.766497 + +epoch 751 +train lattice loss: 0.437993 coord loss: 0.753973 + +epoch 752 +train lattice loss: 0.459618 coord loss: 0.744952 + +epoch 753 +train lattice loss: 0.456425 coord loss: 0.747162 + +epoch 754 +train lattice loss: 0.448306 coord loss: 0.743272 + +epoch 755 +train lattice loss: 0.443921 coord loss: 0.748175 + +epoch 756 +train lattice loss: 0.446669 coord loss: 0.742379 + +epoch 757 +train lattice loss: 0.449125 coord loss: 0.744935 + +epoch 758 +train lattice loss: 0.442462 coord loss: 0.743247 + +epoch 759 +train lattice loss: 0.447616 coord loss: 0.743073 + +epoch 760 +train lattice loss: 0.443254 coord loss: 0.752164 +val lattice loss: 0.445436 coord loss: 0.759078 + +epoch 761 +train lattice loss: 0.442630 coord loss: 0.743805 + +epoch 762 +train lattice loss: 0.448688 coord loss: 0.747498 + +epoch 763 +train lattice loss: 0.443389 coord loss: 0.740601 + +epoch 764 +train lattice loss: 0.441513 coord loss: 0.739396 + +epoch 765 +train lattice loss: 0.441224 coord loss: 0.742382 + +epoch 766 +train lattice loss: 0.445632 coord loss: 0.741104 + +epoch 767 +train lattice loss: 0.451366 coord loss: 0.737595 + +epoch 768 +train lattice loss: 0.453253 coord loss: 0.751117 + +epoch 769 +train lattice loss: 0.447749 coord loss: 0.745416 + +epoch 770 +train lattice loss: 0.451291 coord loss: 0.745400 +val lattice loss: 0.443074 coord loss: 0.735938 +save model with optimal loss + +epoch 771 +train lattice loss: 0.446090 coord loss: 0.744475 + +epoch 772 +train lattice loss: 0.444435 coord loss: 0.746195 + +epoch 773 +train lattice loss: 0.441559 coord loss: 0.739705 + +epoch 774 +train lattice loss: 0.434509 coord loss: 0.745224 + +epoch 775 +train lattice loss: 0.440064 coord loss: 0.738029 + +epoch 776 +train lattice loss: 0.440094 coord loss: 0.739666 + +epoch 777 +train lattice loss: 0.438744 coord loss: 0.737190 + +epoch 778 +train lattice loss: 0.439923 coord loss: 0.741491 + +epoch 779 +train lattice loss: 0.436484 coord loss: 0.743636 + +epoch 780 +train lattice loss: 0.447977 coord loss: 0.736942 +val lattice loss: 0.427041 coord loss: 0.741514 +save model with optimal loss + +epoch 781 +train lattice loss: 0.445513 coord loss: 0.737815 + +epoch 782 +train lattice loss: 0.450998 coord loss: 0.729391 + +epoch 783 +train lattice loss: 0.442136 coord loss: 0.741240 + +epoch 784 +train lattice loss: 0.437858 coord loss: 0.740821 + +epoch 785 +train lattice loss: 0.435571 coord loss: 0.733613 + +epoch 786 +train lattice loss: 0.446861 coord loss: 0.733563 + +epoch 787 +train lattice loss: 0.442779 coord loss: 0.736754 + +epoch 788 +train lattice loss: 0.436168 coord loss: 0.736269 + +epoch 789 +train lattice loss: 0.439674 coord loss: 0.738434 + +epoch 790 +train lattice loss: 0.441835 coord loss: 0.731389 +val lattice loss: 0.444128 coord loss: 0.735422 + +epoch 791 +train lattice loss: 0.443805 coord loss: 0.734845 + +epoch 792 +train lattice loss: 0.446453 coord loss: 0.739031 + +epoch 793 +train lattice loss: 0.445610 coord loss: 0.734502 + +epoch 794 +train lattice loss: 0.443429 coord loss: 0.738874 + +epoch 795 +train lattice loss: 0.439292 coord loss: 0.729878 + +epoch 796 +train lattice loss: 0.442409 coord loss: 0.738992 + +epoch 797 +train lattice loss: 0.436227 coord loss: 0.730820 + +epoch 798 +train lattice loss: 0.436088 coord loss: 0.731266 + +epoch 799 +train lattice loss: 0.435566 coord loss: 0.736791 + +epoch 800 +train lattice loss: 0.442739 coord loss: 0.727269 +val lattice loss: 0.445031 coord loss: 0.736887 + +epoch 801 +train lattice loss: 0.436308 coord loss: 0.737944 + +epoch 802 +train lattice loss: 0.438312 coord loss: 0.728706 + +epoch 803 +train lattice loss: 0.436121 coord loss: 0.723897 + +epoch 804 +train lattice loss: 0.447963 coord loss: 0.725699 + +epoch 805 +train lattice loss: 0.426424 coord loss: 0.736015 + +epoch 806 +train lattice loss: 0.436006 coord loss: 0.725575 + +epoch 807 +train lattice loss: 0.433636 coord loss: 0.726982 + +epoch 808 +train lattice loss: 0.435873 coord loss: 0.726775 + +epoch 809 +train lattice loss: 0.429350 coord loss: 0.731532 + +epoch 810 +train lattice loss: 0.439153 coord loss: 0.720471 +val lattice loss: 0.439110 coord loss: 0.738775 + +epoch 811 +train lattice loss: 0.438699 coord loss: 0.725759 + +epoch 812 +train lattice loss: 0.438656 coord loss: 0.721454 + +epoch 813 +train lattice loss: 0.445314 coord loss: 0.725678 + +epoch 814 +train lattice loss: 0.441471 coord loss: 0.720564 + +epoch 815 +train lattice loss: 0.443261 coord loss: 0.723326 + +epoch 816 +train lattice loss: 0.447354 coord loss: 0.727103 + +epoch 817 +train lattice loss: 0.446462 coord loss: 0.725618 + +epoch 818 +train lattice loss: 0.440385 coord loss: 0.723173 + +epoch 819 +train lattice loss: 0.439617 coord loss: 0.726050 + +epoch 820 +train lattice loss: 0.436946 coord loss: 0.724936 +val lattice loss: 0.441247 coord loss: 0.720554 +save model with optimal loss + +epoch 821 +train lattice loss: 0.435163 coord loss: 0.721765 + +epoch 822 +train lattice loss: 0.436849 coord loss: 0.722679 + +epoch 823 +train lattice loss: 0.442346 coord loss: 0.719029 + +epoch 824 +train lattice loss: 0.441783 coord loss: 0.725787 + +epoch 825 +train lattice loss: 0.435278 coord loss: 0.725479 + +epoch 826 +train lattice loss: 0.432447 coord loss: 0.719428 + +epoch 827 +train lattice loss: 0.430865 coord loss: 0.723928 + +epoch 828 +train lattice loss: 0.437283 coord loss: 0.716028 + +epoch 829 +train lattice loss: 0.429800 coord loss: 0.721686 + +epoch 830 +train lattice loss: 0.433206 coord loss: 0.724807 +val lattice loss: 0.445677 coord loss: 0.731583 + +epoch 831 +train lattice loss: 0.434498 coord loss: 0.715563 + +epoch 832 +train lattice loss: 0.428337 coord loss: 0.720870 + +epoch 833 +train lattice loss: 0.433792 coord loss: 0.721419 + +epoch 834 +train lattice loss: 0.435336 coord loss: 0.723985 + +epoch 835 +train lattice loss: 0.433155 coord loss: 0.715947 + +epoch 836 +train lattice loss: 0.438018 coord loss: 0.715551 + +epoch 837 +train lattice loss: 0.434161 coord loss: 0.724076 + +epoch 838 +train lattice loss: 0.433293 coord loss: 0.718781 + +epoch 839 +train lattice loss: 0.439809 coord loss: 0.715242 + +epoch 840 +train lattice loss: 0.434105 coord loss: 0.715038 +val lattice loss: 0.438590 coord loss: 0.722975 +save model with optimal loss + +epoch 841 +train lattice loss: 0.433369 coord loss: 0.712448 + +epoch 842 +train lattice loss: 0.439253 coord loss: 0.711719 + +epoch 843 +train lattice loss: 0.433210 coord loss: 0.724631 + +epoch 844 +train lattice loss: 0.443007 coord loss: 0.708889 + +epoch 845 +train lattice loss: 0.441152 coord loss: 0.717511 + +epoch 846 +train lattice loss: 0.431136 coord loss: 0.719687 + +epoch 847 +train lattice loss: 0.441349 coord loss: 0.708946 + +epoch 848 +train lattice loss: 0.431030 coord loss: 0.713417 + +epoch 849 +train lattice loss: 0.432541 coord loss: 0.714748 + +epoch 850 +train lattice loss: 0.430578 coord loss: 0.716111 +val lattice loss: 0.427124 coord loss: 0.725963 +save model with optimal loss + +epoch 851 +train lattice loss: 0.437414 coord loss: 0.714072 + +epoch 852 +train lattice loss: 0.421652 coord loss: 0.711217 + +epoch 853 +train lattice loss: 0.425372 coord loss: 0.718088 + +epoch 854 +train lattice loss: 0.431661 coord loss: 0.716808 + +epoch 855 +train lattice loss: 0.427875 coord loss: 0.711581 + +epoch 856 +train lattice loss: 0.432110 coord loss: 0.706073 + +epoch 857 +train lattice loss: 0.424212 coord loss: 0.711312 + +epoch 858 +train lattice loss: 0.438462 coord loss: 0.709573 + +epoch 859 +train lattice loss: 0.427674 coord loss: 0.709172 + +epoch 860 +train lattice loss: 0.431119 coord loss: 0.707033 +val lattice loss: 0.422137 coord loss: 0.717954 +save model with optimal loss + +epoch 861 +train lattice loss: 0.437514 coord loss: 0.709446 + +epoch 862 +train lattice loss: 0.428143 coord loss: 0.701311 + +epoch 863 +train lattice loss: 0.425539 coord loss: 0.705180 + +epoch 864 +train lattice loss: 0.432172 coord loss: 0.700293 + +epoch 865 +train lattice loss: 0.426243 coord loss: 0.707013 + +epoch 866 +train lattice loss: 0.435343 coord loss: 0.710492 + +epoch 867 +train lattice loss: 0.429295 coord loss: 0.708538 + +epoch 868 +train lattice loss: 0.439748 coord loss: 0.711605 + +epoch 869 +train lattice loss: 0.442045 coord loss: 0.709622 + +epoch 870 +train lattice loss: 0.430555 coord loss: 0.713538 +val lattice loss: 0.431699 coord loss: 0.717041 + +epoch 871 +train lattice loss: 0.436831 coord loss: 0.704418 + +epoch 872 +train lattice loss: 0.435518 coord loss: 0.704686 + +epoch 873 +train lattice loss: 0.436000 coord loss: 0.710259 + +epoch 874 +train lattice loss: 0.429730 coord loss: 0.708462 + +epoch 875 +train lattice loss: 0.424293 coord loss: 0.702715 + +epoch 876 +train lattice loss: 0.429878 coord loss: 0.707743 + +epoch 877 +train lattice loss: 0.431091 coord loss: 0.704344 + +epoch 878 +train lattice loss: 0.417228 coord loss: 0.706351 + +epoch 879 +train lattice loss: 0.424031 coord loss: 0.707927 + +epoch 880 +train lattice loss: 0.434224 coord loss: 0.703328 +val lattice loss: 0.430088 coord loss: 0.715569 + +epoch 881 +train lattice loss: 0.418903 coord loss: 0.699489 + +epoch 882 +train lattice loss: 0.428857 coord loss: 0.702321 + +epoch 883 +train lattice loss: 0.429331 coord loss: 0.710992 + +epoch 884 +train lattice loss: 0.438073 coord loss: 0.698795 + +epoch 885 +train lattice loss: 0.429331 coord loss: 0.697728 + +epoch 886 +train lattice loss: 0.427734 coord loss: 0.697467 + +epoch 887 +train lattice loss: 0.419435 coord loss: 0.706503 + +epoch 888 +train lattice loss: 0.423947 coord loss: 0.711391 + +epoch 889 +train lattice loss: 0.424772 coord loss: 0.703549 + +epoch 890 +train lattice loss: 0.424738 coord loss: 0.705284 +val lattice loss: 0.439644 coord loss: 0.695873 +save model with optimal loss + +epoch 891 +train lattice loss: 0.427770 coord loss: 0.709441 + +epoch 892 +train lattice loss: 0.417720 coord loss: 0.695873 + +epoch 893 +train lattice loss: 0.423323 coord loss: 0.695268 + +epoch 894 +train lattice loss: 0.421287 coord loss: 0.691716 + +epoch 895 +train lattice loss: 0.425702 coord loss: 0.699524 + +epoch 896 +train lattice loss: 0.432707 coord loss: 0.699153 + +epoch 897 +train lattice loss: 0.436965 coord loss: 0.696508 + +epoch 898 +train lattice loss: 0.422784 coord loss: 0.701050 + +epoch 899 +train lattice loss: 0.431657 coord loss: 0.698847 + +epoch 900 +train lattice loss: 0.421638 coord loss: 0.699138 +val lattice loss: 0.426740 coord loss: 0.700001 +save model with optimal loss + +epoch 901 +train lattice loss: 0.427483 coord loss: 0.699043 + +epoch 902 +train lattice loss: 0.425018 coord loss: 0.695744 + +epoch 903 +train lattice loss: 0.424966 coord loss: 0.692588 + +epoch 904 +train lattice loss: 0.429186 coord loss: 0.705168 + +epoch 905 +train lattice loss: 0.419077 coord loss: 0.699471 + +epoch 906 +train lattice loss: 0.421136 coord loss: 0.697039 + +epoch 907 +train lattice loss: 0.432549 coord loss: 0.696958 + +epoch 908 +train lattice loss: 0.418046 coord loss: 0.686523 + +epoch 909 +train lattice loss: 0.426404 coord loss: 0.695870 + +epoch 910 +train lattice loss: 0.422402 coord loss: 0.697757 +val lattice loss: 0.420955 coord loss: 0.708106 + +epoch 911 +train lattice loss: 0.419841 coord loss: 0.694849 + +epoch 912 +train lattice loss: 0.424669 coord loss: 0.690866 + +epoch 913 +train lattice loss: 0.421760 coord loss: 0.704045 + +epoch 914 +train lattice loss: 0.413536 coord loss: 0.698825 + +epoch 915 +train lattice loss: 0.416036 coord loss: 0.700281 + +epoch 916 +train lattice loss: 0.414834 coord loss: 0.700496 + +epoch 917 +train lattice loss: 0.429059 coord loss: 0.695338 + +epoch 918 +train lattice loss: 0.424619 coord loss: 0.696612 + +epoch 919 +train lattice loss: 0.427511 coord loss: 0.688693 + +epoch 920 +train lattice loss: 0.423104 coord loss: 0.686626 +val lattice loss: 0.422228 coord loss: 0.698384 +save model with optimal loss + +epoch 921 +train lattice loss: 0.429287 coord loss: 0.687455 + +epoch 922 +train lattice loss: 0.430136 coord loss: 0.691890 + +epoch 923 +train lattice loss: 0.423160 coord loss: 0.687718 + +epoch 924 +train lattice loss: 0.425536 coord loss: 0.686596 + +epoch 925 +train lattice loss: 0.425280 coord loss: 0.693656 + +epoch 926 +train lattice loss: 0.427478 coord loss: 0.685268 + +epoch 927 +train lattice loss: 0.415336 coord loss: 0.690853 + +epoch 928 +train lattice loss: 0.421458 coord loss: 0.687901 + +epoch 929 +train lattice loss: 0.418312 coord loss: 0.689388 + +epoch 930 +train lattice loss: 0.420190 coord loss: 0.696512 +val lattice loss: 0.417773 coord loss: 0.679878 +save model with optimal loss + +epoch 931 +train lattice loss: 0.434790 coord loss: 0.684247 + +epoch 932 +train lattice loss: 0.430169 coord loss: 0.679415 + +epoch 933 +train lattice loss: 0.421866 coord loss: 0.687110 + +epoch 934 +train lattice loss: 0.423605 coord loss: 0.687179 + +epoch 935 +train lattice loss: 0.425403 coord loss: 0.682173 + +epoch 936 +train lattice loss: 0.427836 coord loss: 0.685400 + +epoch 937 +train lattice loss: 0.427026 coord loss: 0.692783 + +epoch 938 +train lattice loss: 0.422625 coord loss: 0.685916 + +epoch 939 +train lattice loss: 0.416242 coord loss: 0.685157 + +epoch 940 +train lattice loss: 0.420131 coord loss: 0.682312 +val lattice loss: 0.419265 coord loss: 0.682671 + +epoch 941 +train lattice loss: 0.417348 coord loss: 0.693279 + +epoch 942 +train lattice loss: 0.420905 coord loss: 0.692588 + +epoch 943 +train lattice loss: 0.414645 coord loss: 0.694753 + +epoch 944 +train lattice loss: 0.417368 coord loss: 0.686060 + +epoch 945 +train lattice loss: 0.417557 coord loss: 0.686836 + +epoch 946 +train lattice loss: 0.428913 coord loss: 0.680269 + +epoch 947 +train lattice loss: 0.423012 coord loss: 0.673449 + +epoch 948 +train lattice loss: 0.426269 coord loss: 0.685075 + +epoch 949 +train lattice loss: 0.416344 coord loss: 0.687929 + +epoch 950 +train lattice loss: 0.425047 coord loss: 0.682253 +val lattice loss: 0.423472 coord loss: 0.690464 + +epoch 951 +train lattice loss: 0.425406 coord loss: 0.684545 + +epoch 952 +train lattice loss: 0.419302 coord loss: 0.680366 + +epoch 953 +train lattice loss: 0.416923 coord loss: 0.687395 + +epoch 954 +train lattice loss: 0.412492 coord loss: 0.684400 + +epoch 955 +train lattice loss: 0.418869 coord loss: 0.683663 + +epoch 956 +train lattice loss: 0.405636 coord loss: 0.691040 + +epoch 957 +train lattice loss: 0.411248 coord loss: 0.673192 + +epoch 958 +train lattice loss: 0.411240 coord loss: 0.674224 + +epoch 959 +train lattice loss: 0.408995 coord loss: 0.689520 + +epoch 960 +train lattice loss: 0.417619 coord loss: 0.684608 +val lattice loss: 0.414132 coord loss: 0.670716 +save model with optimal loss + +epoch 961 +train lattice loss: 0.414883 coord loss: 0.685626 + +epoch 962 +train lattice loss: 0.414092 coord loss: 0.680742 + +epoch 963 +train lattice loss: 0.434757 coord loss: 0.673474 + +epoch 964 +train lattice loss: 0.424048 coord loss: 0.680345 + +epoch 965 +train lattice loss: 0.428207 coord loss: 0.681877 + +epoch 966 +train lattice loss: 0.414209 coord loss: 0.682982 + +epoch 967 +train lattice loss: 0.420112 coord loss: 0.682192 + +epoch 968 +train lattice loss: 0.416650 coord loss: 0.671326 + +epoch 969 +train lattice loss: 0.413828 coord loss: 0.677852 + +epoch 970 +train lattice loss: 0.417996 coord loss: 0.679620 +val lattice loss: 0.411966 coord loss: 0.683468 + +epoch 971 +train lattice loss: 0.420269 coord loss: 0.679800 + +epoch 972 +train lattice loss: 0.417725 coord loss: 0.678362 + +epoch 973 +train lattice loss: 0.416797 coord loss: 0.680872 + +epoch 974 +train lattice loss: 0.411123 coord loss: 0.680860 + +epoch 975 +train lattice loss: 0.411901 coord loss: 0.677532 + +epoch 976 +train lattice loss: 0.417096 coord loss: 0.677052 + +epoch 977 +train lattice loss: 0.413131 coord loss: 0.670863 + +epoch 978 +train lattice loss: 0.416549 coord loss: 0.676844 + +epoch 979 +train lattice loss: 0.413295 coord loss: 0.669396 + +epoch 980 +train lattice loss: 0.412826 coord loss: 0.674910 +val lattice loss: 0.411604 coord loss: 0.688421 + +epoch 981 +train lattice loss: 0.412524 coord loss: 0.671906 + +epoch 982 +train lattice loss: 0.410007 coord loss: 0.673043 + +epoch 983 +train lattice loss: 0.421162 coord loss: 0.670106 + +epoch 984 +train lattice loss: 0.418265 coord loss: 0.673351 + +epoch 985 +train lattice loss: 0.415792 coord loss: 0.675585 + +epoch 986 +train lattice loss: 0.411508 coord loss: 0.668131 + +epoch 987 +train lattice loss: 0.410874 coord loss: 0.671801 + +epoch 988 +train lattice loss: 0.408191 coord loss: 0.671852 + +epoch 989 +train lattice loss: 0.420427 coord loss: 0.672009 + +epoch 990 +train lattice loss: 0.426513 coord loss: 0.672927 +val lattice loss: 0.432883 coord loss: 0.680245 + +epoch 991 +train lattice loss: 0.407853 coord loss: 0.667964 + +epoch 992 +train lattice loss: 0.409659 coord loss: 0.674617 + +epoch 993 +train lattice loss: 0.415244 coord loss: 0.671504 + +epoch 994 +train lattice loss: 0.417565 coord loss: 0.665344 + +epoch 995 +train lattice loss: 0.413815 coord loss: 0.668616 + +epoch 996 +train lattice loss: 0.419454 coord loss: 0.661534 + +epoch 997 +train lattice loss: 0.412735 coord loss: 0.681498 + +epoch 998 +train lattice loss: 0.411767 coord loss: 0.677848 + +epoch 999 +train lattice loss: 0.414840 coord loss: 0.668317 + +epoch 1000 +train lattice loss: 0.419976 coord loss: 0.669637 +val lattice loss: 0.421318 coord loss: 0.666624 + + +save model in the last epoch + +Loading ... +batch 0 / 4, sample 0 / 1 + 0%| | 0/1000 [00:00