benvigano / burner-process Goto Github PK

from scriptifier import *
def func_1(model):
history=model.fit_generator(
train_generator,
steps_per_epoch=train_generator.n//train_generator.batch_size,
epochs=20,initial_epoch=0,
validation_data=validation_generator,
validation_steps=validation_generator.n//validation_generator.batch_size,callbacks=[csv_logger,tb_cb,tb_cb1])
return model,history
scriptified_func_1 = scriptifier.run_as_script()
out = scriptified_func_1(model)

Good job. I encountered the same issue as several before me when trying to fit and predict tensorflow models in a loop. Your tool safed me, but unfortunately does not support thread safety. I fixed that by using temp directories instead of using your higher identifier logic, but have not been able to push my fix to your repo.