HowToSD-ComfyUI-Pt-Wrapper/examples/workflows/train_model_with_custom_loss_example.json

{"last_node_id":95,"last_link_id":119,"nodes":[{"id":61,"type":"PtvDataset","pos":[2007.994873046875,1723.90625],"size":[400,200],"flags":{},"order":10,"mode":0,"inputs":[{"name":"transform","localized_name":"transform","type":"PTVTRANSFORM","shape":7,"link":103}],"outputs":[{"name":"PTVDATASET","localized_name":"PTVDATASET","type":"PTVDATASET","links":[74,83],"slot_index":0}],"properties":{"Node name for S&R":"PtvDataset"},"widgets_values":["CIFAR10",true,"","{\"train\":True}"]},{"id":19,"type":"PtSaveModel","pos":[3530.668701171875,1647.40185546875],"size":[396.8008117675781,78],"flags":{},"order":21,"mode":0,"inputs":[{"name":"model","localized_name":"model","type":"PTMODEL","link":119},{"name":"model_path","type":"STRING","widget":{"name":"model_path"},"link":99}],"outputs":[{"name":"model full path","localized_name":"model full path","type":"STRING","links":[70],"slot_index":0}],"properties":{"Node name for S&R":"PtSaveModel"},"widgets_values":["/container/transfer/models/cifar10.pt"]},{"id":62,"type":"PtvDataset","pos":[2007.994873046875,2039.8226318359375],"size":[400,200],"flags":{},"order":11,"mode":0,"inputs":[{"name":"transform","localized_name":"transform","type":"PTVTRANSFORM","shape":7,"link":89}],"outputs":[{"name":"PTVDATASET","localized_name":"PTVDATASET","type":"PTVDATASET","links":[75,86],"slot_index":0}],"properties":{"Node name for S&R":"PtvDataset"},"widgets_values":["CIFAR10",true,"","{\"train\":False}"]},{"id":76,"type":"PtvDatasetLen","pos":[2007.994873046875,1638.4072265625],"size":[210,26],"flags":{},"order":14,"mode":0,"inputs":[{"name":"dataset","localized_name":"dataset","type":"PTVDATASET","link":83}],"outputs":[{"name":"INT","localized_name":"INT","type":"INT","links":[84],"slot_index":0}],"properties":{"Node name for S&R":"PtvDatasetLen"},"widgets_values":[]},{"id":78,"type":"PtvDatasetLen","pos":[2007.994873046875,2297.724365234375],"size":[210,26],"flags":{},"order":16,"mode":0,"inputs":[{"name":"dataset","localized_name":"dataset","type":"PTVDATASET","link":86}],"outputs":[{"name":"INT","localized_name":"INT","type":"INT","links":[85],"slot_index":0}],"properties":{"Node name for S&R":"PtvDatasetLen"},"widgets_values":[]},{"id":79,"type":"CDAShowInt","pos":[2007.994873046875,2381.43359375],"size":[315,112],"flags":{},"order":19,"mode":0,"inputs":[{"name":"int_scalar","type":"INT","widget":{"name":"int_scalar"},"link":85}],"outputs":[{"name":"STRING","localized_name":"STRING","type":"STRING","shape":6,"links":null}],"properties":{"Node name for S&R":"CDAShowInt"},"widgets_values":[0,"10000"]},{"id":53,"type":"MarkdownNote","pos":[1215.7994384765625,1723.90625],"size":[210,88],"flags":{},"order":0,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Load train data"],"color":"#432","bgcolor":"#653"},{"id":54,"type":"MarkdownNote","pos":[1214.4312744140625,2039.8226318359375],"size":[210,88],"flags":{},"order":1,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Load val data"],"color":"#432","bgcolor":"#653"},{"id":77,"type":"CDAShowInt","pos":[2007.994873046875,1504.9381103515625],"size":[315,112],"flags":{},"order":18,"mode":0,"inputs":[{"name":"int_scalar","type":"INT","widget":{"name":"int_scalar"},"link":84}],"outputs":[{"name":"STRING","localized_name":"STRING","type":"STRING","shape":6,"links":null}],"properties":{"Node name for S&R":"CDAShowInt"},"widgets_values":[0,"50000"]},{"id":90,"type":"PtvTransformsDataAugment","pos":[1595.48486328125,1723.90625],"size":[340.20001220703125,202],"flags":{},"order":2,"mode":0,"inputs":[],"outputs":[{"name":"PTVTRANSFORM","localized_name":"PTVTRANSFORM","type":"PTVTRANSFORM","links":[103],"slot_index":0}],"properties":{"Node name for S&R":"PtvTransformsDataAugment"},"widgets_values":[0.5,0,0,0.125,0.125,1,1]},{"id":81,"type":"PtvTransformsToTensor","pos":[1595.48486328125,2039.8226318359375],"size":[338.8136291503906,26],"flags":{},"order":3,"mode":0,"inputs":[],"outputs":[{"name":"PTVTRANSFORM","localized_name":"PTVTRANSFORM","type":"PTVTRANSFORM","links":[89],"slot_index":0}],"properties":{"Node name for S&R":"PtvTransformsToTensor"},"widgets_values":[]},{"id":56,"type":"MarkdownNote","pos":[3535.058349609375,1775.04931640625],"size":[210,88],"flags":{},"order":4,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Save model"],"color":"#432","bgcolor":"#653"},{"id":80,"type":"PtnResnetModel","pos":[2478.35009765625,1413.751953125],"size":[379.3387145996094,113.2640609741211],"flags":{},"order":5,"mode":0,"inputs":[],"outputs":[{"name":"PTMODEL","localized_name":"PTMODEL","type":"PTMODEL","links":[91,113],"slot_index":0}],"properties":{"Node name for S&R":"PtnResnetModel"},"widgets_values":["(3,32,32)",10,10]},{"id":63,"type":"PtDataLoader","pos":[2473.048828125,1723.90625],"size":[400,200],"flags":{},"order":13,"mode":0,"inputs":[{"name":"dataset","localized_name":"dataset","type":"PTVDATASET","link":74}],"outputs":[{"name":"PTDATALOADER","localized_name":"PTDATALOADER","type":"PTDATALOADER","links":[114],"slot_index":0}],"properties":{"Node name for S&R":"PtDataLoader"},"widgets_values":[128,true,"{\"num_workers\":1}"]},{"id":11,"type":"PtoAdam","pos":[2976.371337890625,1387.1434326171875],"size":[246.16842651367188,110.40398406982422],"flags":{},"order":12,"mode":0,"inputs":[{"name":"model","localized_name":"model","type":"PTMODEL","link":91}],"outputs":[{"name":"PTOPTIMIZER","localized_name":"PTOPTIMIZER","type":"PTOPTIMIZER","links":[111,115],"slot_index":0}],"properties":{"Node name for S&R":"PtoAdam"},"widgets_values":[0.001,0.9,0.999]},{"id":64,"type":"PtDataLoader","pos":[2473.048828125,2039.8226318359375],"size":[400,200],"flags":{},"order":15,"mode":0,"inputs":[{"name":"dataset","localized_name":"dataset","type":"PTVDATASET","link":75}],"outputs":[{"name":"PTDATALOADER","localized_name":"PTDATALOADER","type":"PTDATALOADER","links":[117],"slot_index":0}],"properties":{"Node name for S&R":"PtDataLoader"},"widgets_values":[128,false,"{\"num_workers\":1}"]},{"id":95,"type":"PtnCrossEntropyLoss","pos":[2481.095947265625,1593.5537109375],"size":[368.3888244628906,61.8134880065918],"flags":{},"order":6,"mode":0,"inputs":[],"outputs":[{"name":"PTLOSS","localized_name":"PTLOSS","type":"PTLOSS","links":[116],"slot_index":0}],"properties":{"Node name for S&R":"PtnCrossEntropyLoss"},"widgets_values":["mean"]},{"id":93,"type":"PtoLrSchedulerCosineAnnealing","pos":[3274.407958984375,1388.4407958984375],"size":[280.896240234375,86.71690368652344],"flags":{},"order":17,"mode":0,"inputs":[{"name":"optimizer","localized_name":"optimizer","type":"PTOPTIMIZER","link":111}],"outputs":[{"name":"PTLRSCHEDULER","localized_name":"PTLRSCHEDULER","type":"PTLRSCHEDULER","links":[118],"slot_index":0}],"properties":{"Node name for S&R":"PtoLrSchedulerCosineAnnealing"},"widgets_values":[10,0]},{"id":51,"type":"PandasShowText","pos":[3540.85546875,1921.258056640625],"size":[333.6095275878906,112],"flags":{},"order":22,"mode":0,"inputs":[{"name":"text","type":"STRING","widget":{"name":"text"},"link":70}],"outputs":[{"name":"STRING","localized_name":"STRING","type":"STRING","shape":6,"links":null}],"properties":{"Node name for S&R":"PandasShowText"},"widgets_values":["","/container/transfer/models/cifar10_tmp.pt"]},{"id":55,"type":"MarkdownNote","pos":[2988.48876953125,1922.287353515625],"size":[232.82992553710938,88],"flags":{},"order":7,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Train"],"color":"#432","bgcolor":"#653"},{"id":52,"type":"MarkdownNote","pos":[2484.47021484375,1271.1656494140625],"size":[210,88],"flags":{},"order":8,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Define model"],"color":"#432","bgcolor":"#653"},{"id":89,"type":"CDAStringCreate","pos":[3536.05712890625,1534.6680908203125],"size":[385.5651550292969,59.037696838378906],"flags":{},"order":9,"mode":0,"inputs":[],"outputs":[{"name":"STRING","localized_name":"STRING","type":"STRING","links":[99],"slot_index":0}],"properties":{"Node name for S&R":"CDAStringCreate"},"widgets_values":["/container/transfer/models/cifar10_tmp.pt"]},{"id":94,"type":"PtTrainModel","pos":[2985.846923828125,1603.26904296875],"size":[315,254],"flags":{},"order":20,"mode":0,"inputs":[{"name":"model","localized_name":"model","type":"PTMODEL","link":113},{"name":"train_loader","localized_name":"train_loader","type":"PTDATALOADER","link":114},{"name":"optimizer","localized_name":"optimizer","type":"PTOPTIMIZER","link":115},{"name":"loss_function","localized_name":"loss_function","type":"PTLOSS","link":116},{"name":"scheduler","localized_name":"scheduler","type":"PTLRSCHEDULER","shape":7,"link":118},{"name":"val_loader","localized_name":"val_loader","type":"PTDATALOADER","shape":7,"link":117}],"outputs":[{"name":"Model","localized_name":"Model","type":"PTMODEL","links":[119],"slot_index":0},{"name":"train loss","localized_name":"train loss","type":"TENSOR","links":null},{"name":"val loss","localized_name":"val loss","type":"TENSOR","links":null}],"properties":{"Node name for S&R":"PtTrainModel"},"widgets_values":[10,true,false,10,false]}],"links":[[70,19,0,51,0,"STRING"],[74,61,0,63,0,"PTVDATASET"],[75,62,0,64,0,"PTVDATASET"],[83,61,0,76,0,"PTVDATASET"],[84,76,0,77,0,"INT"],[85,78,0,79,0,"INT"],[86,62,0,78,0,"PTVDATASET"],[89,81,0,62,0,"PTVTRANSFORM"],[91,80,0,11,0,"PTMODEL"],[99,89,0,19,1,"STRING"],[103,90,0,61,0,"PTVTRANSFORM"],[111,11,0,93,0,"PTOPTIMIZER"],[113,80,0,94,0,"PTMODEL"],[114,63,0,94,1,"PTDATALOADER"],[115,11,0,94,2,"PTOPTIMIZER"],[116,95,0,94,3,"PTLOSS"],[117,64,0,94,5,"PTDATALOADER"],[118,93,0,94,4,"PTLRSCHEDULER"],[119,94,0,19,0,"PTMODEL"]],"groups":[],"config":{},"extra":{"ds":{"scale":0.9598170997212797,"offset":[-2396.5953376811794,-1204.7438265178134]},"node_versions":{"pt-wrapper":"f42fef60b92ca64335192e9c97b23fe85f23b0ad","data-analysis":"68dd090367adffa569c3b3993d56e422f4516d2d"},"ue_links":[]},"version":0.4}