cyto_dl.callbacks.csv_saver module#

class cyto_dl.callbacks.csv_saver.CSVSaver(save_dir, meta_keys=[])[source]#

Bases: Callback

on_predict_epoch_end(trainer, pl_module)[source]#
pred_to_csv(pred)[source]#
save_feats(predictions, stage)[source]#
class cyto_dl.callbacks.csv_saver.JEPASaver(save_dir, meta_keys=[])[source]#

Bases: CSVSaver

pred_to_csv(pred)[source]#