SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

add distant debugging

Browse Source
This commit is contained in:
thomwolf
2019-02-08 11:03:59 +01:00
parent 5ee4f17234
commit 5adc20723b
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/run_openai_gpt.py
View File

@@ -99,9 +99,21 @@ def main():
parser.add_argument('--weight_decay', type=float, default=0.01) parser.add_argument('--weight_decay', type=float, default=0.01)
parser.add_argument('--lm_coef', type=float, default=0.5) parser.add_argument('--lm_coef', type=float, default=0.5)
parser.add_argument('--n_valid', type=int, default=374) parser.add_argument('--n_valid', type=int, default=374)
parser.add_argument('--server_ip', type=str, default='')
parser.add_argument('--server_port', type=str, default='')
args = parser.parse_args() args = parser.parse_args()
print(args) print(args)
# Some distant debugging
# See https://code.visualstudio.com/docs/python/debugging#_attach-to-a-local-script
import ptvsd
print(sys.argv)
print("Waiting for debugger attach")
ptvsd.enable_attach(address=(args.server_ip, args.server_port), redirect_output=True)
ptvsd.wait_for_attach()
random.seed(args.seed) random.seed(args.seed)
np.random.seed(args.seed) np.random.seed(args.seed)
torch.manual_seed(args.seed) torch.manual_seed(args.seed)