Functions — github.com/ShomyLiu/torch-ddp-examples

CodeHub MCPcopy Create free account

hub / github.com/ShomyLiu/torch-ddp-examples / functions

Functions15 in github.com/ShomyLiu/torch-ddp-examples

⨍ Functions 15 ◇ Types & classes 1

↓ 4 callersFunctionload_data_and_labels

Loads MR polarity data from files, splits the data into words and generates labels. Returns split sentences and labels.

↓ 1 callersFunctionclean_str

Tokenization/string cleaning for all datasets except for SST. Original taken from https://github.com/yoonkim/CNN_sentence/blob/master/process

↓ 1 callersFunctiontest

(model, test_loader, device)

horovod_main.py:79

↓ 1 callersFunctiontest

(model, test_loader)

accelerate_main.py:60

↓ 1 callersFunctiontest

(model, test_loader, device)

↓ 1 callersFunctiontest

(model, test_loader, device)

↓ 1 callersFunctiontrain

horovod_main.py:23

↓ 1 callersFunctiontrain

accelerate_main.py:19

↓ 1 callersFunctiontrain

↓ 1 callersFunctiontrain

Method__getitem__

Functioncollate_fn