low_rank_comparisons/src/gpt2_beam.py [19:30]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
torch.set_printoptions(threshold=100000)

import numpy as np

from gpu import (
    add_gpu_params, 
    parse_gpu, 
    distributed_opt, 
    distributed_gather, 
    distributed_sync, 
    cleanup
)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



low_rank_comparisons/src/gpt2_ft.py [14:23]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
torch.set_printoptions(threshold=100000)

from gpu import (
    add_gpu_params, 
    parse_gpu, 
    distributed_opt, 
    distributed_gather, 
    distributed_sync, 
    cleanup
)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -