hhk7734/tensorflow-yolov4

Test steps_per_execution

Closed this issue · 1 comments

def compile(
self,
loss_iou_type: str = "ciou",
loss_verbose=1,
optimizer=optimizers.Adam(learning_rate=1e-4),
**kwargs
):
# TODO: steps_per_execution tensorflow2.4.0-rc4
self.model.compile(
optimizer=optimizer,
loss=train.YOLOv4Loss(
batch_size=self.batch_size,
iou_type=loss_iou_type,
verbose=loss_verbose,
),
**kwargs
)

tested