Skip to content

Commit b87a1ea

Browse files
authored
fix ema copy problem and correct config of ppyolo in static graph (#3371)
1 parent 27d4a24 commit b87a1ea

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

ppdet/engine/trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717
from __future__ import print_function
1818

1919
import os
20+
import sys
21+
import copy
2022
import time
2123
import random
2224
import datetime
@@ -326,7 +328,7 @@ def train(self, validate=False):
326328

327329
# apply ema weight on model
328330
if self.use_ema:
329-
weight = self.model.state_dict()
331+
weight = copy.deepcopy(self.model.state_dict())
330332
self.model.set_dict(self.ema.apply())
331333

332334
self._compose_callback.on_epoch_end(self.status)

static/configs/ppyolo/ppyolov2_reader.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ TrainReader:
1717
beta: 1.5
1818
- !ColorDistort {}
1919
- !RandomExpand
20-
ratio: 2.0
2120
fill_value: [123.675, 116.28, 103.53]
2221
- !RandomCrop {}
2322
- !RandomFlipImage

0 commit comments

Comments
 (0)