mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-09-19 21:35:58 +00:00
25 lines
571 B
Python
25 lines
571 B
Python
|
import os
|
||
|
|
|
||
|
|
import tensorflow as tf
|
||
|
|
|
||
|
|
def make_callbacks(dirpath):
|
||
|
|
dirpath_checkpoints = os.path.join(dirpath, "checkpoints")
|
||
|
|
filepath_metrics = os.path.join(dirpath, "metrics.tsv")
|
||
|
|
|
||
|
|
if not os.path.exists(dirpath_checkpoints):
|
||
|
|
os.mkdir(dirpath_checkpoints)
|
||
|
|
|
||
|
|
return [
|
||
|
|
tf.keras.callbacks.ModelCheckpoint(
|
||
|
|
filepath=os.path.join(
|
||
|
|
dirpath_checkpoints,
|
||
|
|
"checkpoint_weights_e{epoch:d}_loss{loss:.3f}.hdf5"
|
||
|
|
),
|
||
|
|
monitor="loss"
|
||
|
|
),
|
||
|
|
tf.keras.callbacks.CSVLogger(
|
||
|
|
filename=filepath_metrics,
|
||
|
|
separator="\t"
|
||
|
|
),
|
||
|
|
tf.keras.callbacks.ProgbarLogger()
|
||
|
|
]
|