WandbCallback
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
)