BNN-UPC / ignnition

Framework for fast prototyping of Graph Neural Networks
Apache License 2.0
52 stars 16 forks source link

Epoch Size as optional parameter #119

Closed MiquelFerriol closed 2 years ago

MiquelFerriol commented 2 years ago

Solved a problem where the optional parameter "epoch_size" was not defined. This caused an error since, by default, the tf.dataset was "repeated".