Edit evaluation to use weighted instead as there are imbalance labels bad72d7 jesse-tong commited on Apr 11
Change loss function for multi-category because CrossEntropyLoss only accept outputs of (batch, x) and labels with shape of (batch) cb428cb jesse-tong commited on Apr 6