Commit 304c66f9 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

Camembert correctement gelée

parent dfaf429b
......@@ -33,4 +33,4 @@ export PYTHONPATH="/lium/raid01_b/gcaillaut/polysemy/minibert:${PYTHONPATH}"
set -x
set -e
python train.py camembert-t2 ${TRAIN} ${TEST} ${DEV} --outdir ${OUT_DIR} --bs ${BS} -e 10 --epochs-between-save 1 --logdir ${LOGDIR} --device ${DEVICE}
python train.py camembert-t2 ${TRAIN} ${TEST} ${DEV} --outdir ${OUT_DIR} --bs ${BS} -e 10 --epochs-between-save 1 --logdir ${LOGDIR} --device ${DEVICE} --freeze
......@@ -1035,7 +1035,7 @@ def finetune_t1_camembert(args):
prev_epoch = checkpoint["epoch"]
if args.freeze:
for param in model.roberta.roberta.parameters():
for param in model.roberta.parameters():
param.requires_grad = False
model.train()
......@@ -1340,7 +1340,7 @@ def finetune_t2_camembert(args):
prev_epoch = checkpoint["epoch"]
if args.freeze:
for param in model.roberta.roberta.parameters():
for param in model.roberta.parameters():
param.requires_grad = False
model.train()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment