Competitive-Good4690 OP t1_iv4e66u wrote on November 5, 2022 at 5:12 AM

Reply to comment by sgjoesg in U-Net architecture by Competitive-Good4690

So model.fit is an inbuilt function for training the model which I don’t want to use. I want to define the model on my own

sgjoesg t1_iv4ezuu wrote on November 5, 2022 at 5:22 AM

As far as i know, if you create your own class inheriting from tf.keras, then that class can use .fit function. Eg class unet(tf.keras) Model_definition_as_per_arch model = unet() model.fit(data)

So you have the control to use your own model, and use keras's easy training loop as well.

puppet_pals t1_iv4jtpc wrote on November 5, 2022 at 6:27 AM

>So model.fit is an inbuilt function for training the model which I don’t want to use. I want to define the model on my own

just use the UNet using that runs with model.fit() and implement your own training loop following this guide: https://lukewood.xyz/blog/keras-model-walkthrough

Competitive-Good4690 OP t1_iv4fdo1 wrote on November 5, 2022 at 5:27 AM

Yes thank you I’m Referring to Connor Shorten’s video on U-Net (Keras) he’s saying exactly what u just said.. thank you for the response.. really appreciated.. Abhisek Thakur is using PyTorch but he did explain the concept well

sgjoesg t1_iv4hum4 wrote on November 5, 2022 at 5:59 AM

Glad i could help(in some way lol)💯