logo

8 Commits (87fc5edf834dcc4ce892d8742416543a2d8fb024)

Author SHA1 Message Date
zhang chen 7c78b94de1 add some training examples. 3 years ago
Jael Gu 5341779d16 update 3 years ago
Your Name 12e87d618e modify batch_size 3 years ago
Your Name 0b1571476a Add freeze_bn config 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
zhang chen 28cdc0f73a resnet op train from yaml config. 3 years ago