search

shelljane / lithobench

LithoBench: Benchmarking AI Computational Lithography for Semiconductor Manufacturing
MIT License
38 stars 10 forks source link

Training model on custom dataset #2

Open mskv99 opened 3 months ago

mskv99 commented 3 months ago

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 commented 1 month ago

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