Open mskv99 opened 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 :)
Sorry for replying late. This can be achieved by adding the new model or dataset in lithobench/train.py and lithobench/test.py.
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 :)