logo

6 Commits (ec5832e703c9779c53f37dc5488ca67b13636a11)

Author SHA1 Message Date
Your Name 9b0f12749e update test 3 years ago
zhang chen 086c225df0 fix embedding_extractor.py memory leak problem, and add a `num_classes` param in the construct function of operator. 3 years ago
zhang chen f1346e4f45 add eval ability 3 years ago
zhang chen 51a17a4bb0 add custom optimizer and loss case 3 years ago
Jael Gu 4ae024fdeb Update test.py 3 years ago
zhang chen 28cdc0f73a resnet op train from yaml config. 3 years ago