メインコンテンツまでスキップ

WandbCallback

GitHubでソースを見る

WandbCallbackは自動的にKerasとwandbを統合します。

WandbCallback(
monitor="val_loss", verbose=0, mode="auto", save_weights_only=(False),
log_weights=(False), log_gradients=(False), save_model=(True),
training_data=None, validation_data=None, labels=[], predictions=36,
generator=None, input_type=None, output_type=None, log_evaluation=(False),
validation_steps=None, class_colors=None, log_batch_frequency=None,
log_best_prefix="best_", save_graph=(True), validation_indexes=None,
validation_row_processor=None, prediction_row_processor=None,
infer_missing_processors=(True), log_evaluation_frequency=0,
compute_flops=(False), **kwargs
)

例:

model.fit(
X_train,
y_train,
validation_data=(X_test, y_test),
callbacks=[WandbCallback()],
)

メソッド

get_flops

ソースを見る

get_flops() -> float

推論モードでのtf.keras.Modelまたはtf.keras.SequentialモデルのFLOPS[GFLOPs]を計算します。内部ではtf.compat.v1.profilerを使用しています。

set_model

ソースを見る

set_model(
model
)

set_params

ソースを見る

set_params(
params
)
Was this page helpful?👍👎