logo

7 Commits (0b1571476ad15ac723e23f8ee36155b35c98de57)

Author SHA1 Message Date
Your Name 0b1571476a Add freeze_bn config 3 years ago
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