Skip to content

Commit

Permalink
gitignore for perlmutter
Browse files Browse the repository at this point in the history
  • Loading branch information
Shuai Jiang committed Jun 11, 2024
1 parent 018131e commit 8489b69
Show file tree
Hide file tree
Showing 4 changed files with 14 additions and 5 deletions.
5 changes: 5 additions & 0 deletions examples/bert-training/src/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
*.npy
*.png
*.err
*.out

File renamed without changes.
3 changes: 2 additions & 1 deletion examples/machine_translation/src/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,8 @@

# lang_src, lang_tgt = 'de', 'en'
lang_src, lang_tgt = 'en', 'de'
dir_data = '../data/deen_translation/'
# dir_data = '../data/deen_translation/'
dir_data = ''
if 'data' not in os.listdir('..'): dir_data = '../' + dir_data
fn = {
'train': {
Expand Down
11 changes: 7 additions & 4 deletions examples/morphological_classification/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,8 +101,8 @@ def train(rank, params, model, train_loader, optimizer, epoch, compose, device):

stop_time = timer()
optimizer.step()
# if rank == 0:

# if rank == 0:
# root_print(rank, f'rank{rank}, batch_idx {batch_idx}, data {data}, target {target}, loss {loss}')
# for p in model.parameters(): root_print(rank, f'{p.shape}, {p.ravel()[:10]}')
# sys.exit()
Expand Down Expand Up @@ -220,8 +220,11 @@ def main():

# Finish assembling training and test datasets
root_print(rank, 'Loading dataset')
data_path_train = '/users/msalvado/MLT/ML_PQ/data/en_gum-ud-train.conllu.txt'
data_path_dev = '/users/msalvado/MLT/ML_PQ/data/en_gum-ud-dev.conllu.txt'
# data_path_train = '/users/msalvado/MLT/ML_PQ/data/en_gum-ud-train.conllu.txt'
# data_path_dev = '/users/msalvado/MLT/ML_PQ/data/en_gum-ud-dev.conllu.txt'
data_path_train = 'en_gum-ud-train.conllu.txt'
data_path_dev = 'en_gum-ud-dev.conllu.txt'

train_ds, eval_ds, train_dl, eval_dl, vocabs = obtain_ds_dl(data_path_train, data_path_dev, args.batch_size, max_len=2048)
train_set, test_set, train_loader, test_loader = train_ds, eval_ds, train_dl, eval_dl
# root_print(rank, f'{len(train_loader)}, {next(iter(train_loader))}')
Expand Down

0 comments on commit 8489b69

Please sign in to comment.