Merge branch 'hepj-test' into 'main'

修改VIT模型loss就地操作

See merge request dcutoolkit/deeplearing/dlexamples_new!59

PyTorch/NLP/Vision_Transformer/engine_pretrain.py

@@ -53,7 +53,9 @@ def train_one_epoch(model: torch.nn.Module,
             print("Loss is {}, stopping training".format(loss_value))
             sys.exit(1)
 
-        loss /= accum_iter
+        loss_new= loss/accum_iter
+        loss=loss_new
+
         loss_scaler(loss, optimizer, parameters=model.parameters(),
                     update_grad=(data_iter_step + 1) % accum_iter == 0)
         if (data_iter_step + 1) % accum_iter == 0:
@@ -79,4 +81,4 @@ def train_one_epoch(model: torch.nn.Module,
     # gather the stats from all processes
     metric_logger.synchronize_between_processes()
     print("Averaged stats:", metric_logger)
-    return {k: meter.global_avg for k, meter in metric_logger.meters.items()}
+    return {k: meter.global_avg for k, meter in metric_logger.meters.items()}
\ No newline at end of file