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

{"last_node_id":93,"last_link_id":112,"nodes":[{"id":61,"type":"PtvDataset","pos":[2007.994873046875,1723.90625],"size":[400,200],"flags":{},"order":9,"mode":0,"inputs":[{"name":"transform","type":"PTVTRANSFORM","link":103,"shape":7}],"outputs":[{"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,75.65961456298828],"flags":{},"order":20,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":108},{"name":"model_path","type":"STRING","link":99,"widget":{"name":"model_path"}}],"outputs":[{"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":10,"mode":0,"inputs":[{"name":"transform","type":"PTVTRANSFORM","link":89,"shape":7}],"outputs":[{"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":13,"mode":0,"inputs":[{"name":"dataset","type":"PTVDATASET","link":83}],"outputs":[{"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":15,"mode":0,"inputs":[{"name":"dataset","type":"PTVDATASET","link":86}],"outputs":[{"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,76],"flags":{},"order":18,"mode":0,"inputs":[{"name":"int_scalar","type":"INT","link":85,"widget":{"name":"int_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"CDAShowInt"},"widgets_values":[0,"10000"]},{"id":53,"type":"MarkdownNote","pos":[1215.7994384765625,1723.90625],"size":[210,58.94380569458008],"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,58.94380569458008],"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,76],"flags":{},"order":17,"mode":0,"inputs":[{"name":"int_scalar","type":"INT","link":84,"widget":{"name":"int_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"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","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","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,58.94380569458008],"flags":{},"order":4,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Save model"],"color":"#432","bgcolor":"#653"},{"id":51,"type":"PandasShowText","pos":[3536.080078125,1878.28125],"size":[333.6095275878906,79.25810241699219],"flags":{},"order":21,"mode":0,"inputs":[{"name":"text","type":"STRING","link":70,"widget":{"name":"text"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"PandasShowText"},"widgets_values":["","/container/transfer/models/cifar10_v11_2nd_run.pt"]},{"id":52,"type":"MarkdownNote","pos":[2485.625732421875,1294.277587890625],"size":[210,58.94380569458008],"flags":{},"order":5,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Define model"],"color":"#432","bgcolor":"#653"},{"id":80,"type":"PtnResnetModel","pos":[2478.35009765625,1413.751953125],"size":[379.3387145996094,113.2640609741211],"flags":{},"order":6,"mode":0,"inputs":[],"outputs":[{"name":"PTMODEL","type":"PTMODEL","links":[91,104],"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":12,"mode":0,"inputs":[{"name":"dataset","type":"PTVDATASET","link":74}],"outputs":[{"name":"PTDATALOADER","type":"PTDATALOADER","links":[105],"slot_index":0}],"properties":{"Node name for S&R":"PtDataLoader"},"widgets_values":[128,true,"{\"num_workers\":1}"]},{"id":64,"type":"PtDataLoader","pos":[2473.048828125,2039.8226318359375],"size":[400,200],"flags":{},"order":14,"mode":0,"inputs":[{"name":"dataset","type":"PTVDATASET","link":75}],"outputs":[{"name":"PTDATALOADER","type":"PTDATALOADER","links":[106],"slot_index":0}],"properties":{"Node name for S&R":"PtDataLoader"},"widgets_values":[128,false,"{\"num_workers\":1}"]},{"id":55,"type":"MarkdownNote","pos":[3030.089599609375,1947.71044921875],"size":[232.82992553710938,58],"flags":{},"order":7,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["# Train"],"color":"#432","bgcolor":"#653"},{"id":91,"type":"PtTrainClassificationModelLr","pos":[3023.411376953125,1655.6585693359375],"size":[403.1999816894531,234],"flags":{},"order":19,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":104},{"name":"train_loader","type":"PTDATALOADER","link":105},{"name":"optimizer","type":"PTOPTIMIZER","link":107},{"name":"scheduler","type":"PTLRSCHEDULER","link":112},{"name":"val_loader","type":"PTDATALOADER","link":106,"shape":7}],"outputs":[{"name":"Model","type":"PTMODEL","links":[108],"slot_index":0},{"name":"train loss","type":"TENSOR","links":null},{"name":"val loss","type":"TENSOR","links":null}],"properties":{"Node name for S&R":"PtTrainClassificationModelLr"},"widgets_values":[200,true,false,10,false]},{"id":11,"type":"PtoAdam","pos":[2976.371337890625,1387.1434326171875],"size":[246.16842651367188,110.40398406982422],"flags":{},"order":11,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":91}],"outputs":[{"name":"PTOPTIMIZER","type":"PTOPTIMIZER","links":[107,111],"slot_index":0}],"properties":{"Node name for S&R":"PtoAdam"},"widgets_values":[0.001,0.9,0.999]},{"id":93,"type":"PtoLrSchedulerCosineAnnealing","pos":[3274.407958984375,1388.4407958984375],"size":[280.896240234375,86.71690368652344],"flags":{},"order":16,"mode":0,"inputs":[{"name":"optimizer","type":"PTOPTIMIZER","link":111}],"outputs":[{"name":"PTLRSCHEDULER","type":"PTLRSCHEDULER","links":[112],"slot_index":0}],"properties":{"Node name for S&R":"PtoLrSchedulerCosineAnnealing"},"widgets_values":[200,0]},{"id":89,"type":"CDAStringCreate","pos":[3536.05712890625,1534.6680908203125],"size":[385.5651550292969,59.037696838378906],"flags":{},"order":8,"mode":0,"inputs":[],"outputs":[{"name":"STRING","type":"STRING","links":[99],"slot_index":0}],"properties":{"Node name for S&R":"CDAStringCreate"},"widgets_values":["/container/transfer/models/cifar10_v11_2nd_run.pt"]}],"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"],[104,80,0,91,0,"PTMODEL"],[105,63,0,91,1,"PTDATALOADER"],[106,64,0,91,4,"PTDATALOADER"],[107,11,0,91,2,"PTOPTIMIZER"],[108,91,0,19,0,"PTMODEL"],[111,11,0,93,0,"PTOPTIMIZER"],[112,93,0,91,3,"PTLRSCHEDULER"]],"groups":[],"config":{},"extra":{"ds":{"scale":0.3405679149452571,"offset":[-65.83749226699729,-475.28970703078994]},"node_versions":{"pt-wrapper":"eb9d05ed533f01888758f204deea434badf4a0a9","data-analysis":"bde4a07755390554f858e0b8fd8554a3123a6612"},"ue_links":[]},"version":0.4}