Training model on custom dataset

[mskv99]

Could you tell whether it is possible to train one of already implemented models on custom dataset for mask simulation? if yes, could you provide an instruction or may be an example in collab? it would be super cool :)

[shelljane]

Sorry for replying late. This can be achieved by adding the new model or dataset in lithobench/train.py and lithobench/test.py.

[gaussiangit]

can you provide litho_sim4 from ICCAD 2013 ? how did you get those kernels .pts ? how do we generate them for new dataset ?