o ~å$g{ã@s ddlmZGdd„deƒZdS)é)ÚCallbackcs$eZdZ‡fdd„Zdd„Z‡ZS)ÚIncreaseDataEpochcstƒ ¡dS)N)ÚsuperÚ__init__)Úself©Ú __class__©ú3/home/dufour/Documents/diff_plonk/callbacks/data.pyrszIncreaseDataEpoch.__init__cCs,|j}t|jjdƒr|jjj |¡dSdS)NÚ shared_epoch)Ú current_epochÚhasattrÚ datamoduleÚ train_datasetr Ú set_value)rÚtrainerÚ pl_moduleÚepochr r r Úon_train_epoch_startsÿz&IncreaseDataEpoch.on_train_epoch_start)Ú__name__Ú __module__Ú __qualname__rrÚ __classcell__r r rr rs rN)Úpytorch_lightning.callbacksrrr r r r Ús